- 在.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
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