- 在.py中指定
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "3" #指定第三块gpu
- 在命令行指定
CUDA_VISIBLE_DEVICES = 2, 3, 4 bash your_file.sh
- 在环境变量中指定
#打开.bashrc,#打开.bashrc,加入下面代码
export CUDA_VISIBLE_DEVICES = 2, 3, 4
转载服务器指定GPU跑程序
最后
以上就是丰富电灯胆最近收集整理的关于服务器指定GPU跑程序的全部内容,更多相关服务器指定GPU跑程序内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复