服务器有多台GPU 如何指定GPU运行程序

方法1:直接在运行程序命令 前添加  CUDA_VISIBLE_DEVICES=1(自己服务器上的GPU number) 样例如下：

CUDA_VISIBLE_DEVICES=1 python your_program.py

方法2：在自己的代码中添加

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"