2018年9月13日
[转] pytorch指定GPU
摘要： 查过好几次这个命令，总是忘，转一篇mark一下吧 转自：http://www.cnblogs.com/darkknightzh/p/6836568.html PyTorch默认使用从0开始的GPU，如果GPU0正在运行程序，需要指定其他GPU。 有如下两种方法来指定需要使用的GPU。 1. 类似te 阅读全文
posted @ 2018-09-13 10:34 凌波微步_Arborday 阅读(221) 评论(0) 推荐(0) 编辑