转载请注明出处:

http://www.cnblogs.com/darkknightzh/p/8576825.html

 

如果使用多gpu运行程序,可以直接使用CUDA_VISIBLE_DEVICES=0,1,2,3 python xxx.py来设置该程序可见的gpu。当然也可以在程序开头设置os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3'来使用。

 

如果在pycharm中调试时,使用多gpu的话,除了直接在程序中设置os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3',也可以通过下面的方法:

1. 打开Run/Debug Configurations,在Environment variables右边点击…处。

(原)pycharm中使用CUDA_VISIBLE_DEVICES

2. 点击下图中1处的’+’,会出现2处,增加CUDA_VISIBLE_DEVICES,再点击3处,填入0,1,2,3。(下图中第一行已经增加了,只不过为了说明,因而这样表示)

(原)pycharm中使用CUDA_VISIBLE_DEVICES

3. 在pycharm中正常运行程序,就可以了(为了测试是否正确,实际上使用的是2,3,4,5这4个GPU,显示的正确,说明这样配置有效果)。

(原)pycharm中使用CUDA_VISIBLE_DEVICES

相关文章:

  • 2021-11-16
  • 2021-06-12
  • 2021-10-14
  • 2021-06-04
  • 2021-11-04
  • 2021-10-04
猜你喜欢
  • 2022-12-23
  • 2022-01-29
  • 2022-12-23
  • 2022-02-02
  • 2021-07-08
  • 2021-07-06
相关资源
相似解决方案