方法一:执行程序的时候加入语句
CUDA_VISIBLE_DEVICES=1 python xxx.py
方法二:在python代码加入
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
方法三:改变环境变量
export CUDA_VISIBLE_DEVICES=1
方法一:执行程序的时候加入语句
CUDA_VISIBLE_DEVICES=1 python xxx.py
方法二:在python代码加入
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
方法三:改变环境变量
export CUDA_VISIBLE_DEVICES=1