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

只需要添加如下两行代码： import os os.environ["CUDA_VISIBLE_DEVICES"] = "1" 注意，服务器看到的GPU序号和手动在Linux上查看的序号是相反的，比如  那么我们指定os.environ[“CUDA_VISIBLE_DEVICES”] = “0”的话，你会发现在watch nvidia-smi指令下，2号GPU正在满负荷的运行！ ---------

>>阅读原文<<