2017年5月10日

(原)PyTorch中使用指定的GPU

摘要: 转载请注明出处: http://www.cnblogs.com/darkknightzh/p/6836568.html PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。 有如下两种方法来指定需要使用的GPU。 1. 类似tensorflow指定GPU的方式,使用 阅读全文

posted @ 2017-05-10 16:20 darkknightzh 阅读(155669) 评论(6) 推荐(3) 编辑

导航