方法一:在代码头部加入如下代码:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1" # 填写你要使用的GPU号
方法二:在命令行运行该Python文件,并在头部加上你要选择的GPU号:
方法一:在代码头部加入如下代码:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1" # 填写你要使用的GPU号
方法二:在命令行运行该Python文件,并在头部加上你要选择的GPU号: