如何指定程序在某个GPU上运行

只需要添加如下两行代码：

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

注意，服务器看到的GPU序号和手动在Linux上查看的序号是相反的，比如

那么我们指定os.environ[“CUDA_VISIBLE_DEVICES”] = “0”的话，你会发现在watch nvidia-smi指令下，2号GPU正在满负荷的运行！