- 改变系统变量
改变系统环境变量仅使目标显卡,编辑 .bashrc文件,添加系统变量
export CUDA_VISIBLE_DEVICES=0
- 在程序开头设置
os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3'
- 在运行程序时指定
运行程序时使用命令行,来设置该程序可见的gpu:
CUDA_VISIBLE_DEVICES=0,1,2,3 python xxx.py
改变系统环境变量仅使目标显卡,编辑 .bashrc文件,添加系统变量
export CUDA_VISIBLE_DEVICES=0
os.environ["CUDA_VISIBLE_DEVICES"] = '0,1,2,3'
运行程序时使用命令行,来设置该程序可见的gpu:
CUDA_VISIBLE_DEVICES=0,1,2,3 python xxx.py