方法1:直接在运行程序命令 前添加 CUDA_VISIBLE_DEVICES=1(自己服务器上的GPU number) 样例如下:
CUDA_VISIBLE_DEVICES=1 python your_program.py
方法2:在自己的代码中添加
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
方法1:直接在运行程序命令 前添加 CUDA_VISIBLE_DEVICES=1(自己服务器上的GPU number) 样例如下:
CUDA_VISIBLE_DEVICES=1 python your_program.py
方法2:在自己的代码中添加
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"