Linux | 指定使用的GPU :
方法一:在终端执行程序时指定GPU(推荐)
CUDA_VISIBLE_DEVICES=1 python run_file.py
方法二:在Python代码中指定GPU
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
CUDA_VISIBLE_DEVICES=1 python run_file.py
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"