2018年3月15日

(原)pycharm中使用CUDA_VISIBLE_DEVICES

摘要: 转载请注明出处: http://www.cnblogs.com/darkknightzh/p/8576825.html 如果使用多gpu运行程序,可以直接使用CUDA_VISIBLE_DEVICES=0,1,2,3 python xxx.py来设置该程序可见的gpu。当然也可以在程序开头设置os.e 阅读全文

posted @ 2018-03-15 22:30 darkknightzh 阅读(11924) 评论(0) 推荐(2) 编辑

导航