如果使用多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’,也可以通过下面的方法:
- 打开Run/Debug Configurations,在Environment variables右边点击…处。


本文介绍了如何在Python程序中利用CUDA_VISIBLE_DEVICES设置多GPU环境,包括直接命令行方式和PyCharm调试中的环境变量配置。重点讲解了在PyCharm中使用多GPU的实用技巧。
2万+

被折叠的 条评论
为什么被折叠?



