pytorch 问题总结

发表于 更新于 本文字数： 141 阅读时长 ≈ 1 分钟
总结在使用pytorch时遇到的问题

如何指定gpu的使用？

在一个机子上有多个gpu，若要指定某个/多个gpu单独跑一个程序，该怎么办

参考：https://discuss.pytorch.org/t/how-to-change-the-default-device-of-gpu-device-ids-0/1041/25