如何指定程序在某個GPU上運行 python

只需要添加如下兩行代碼： import os os.environ["CUDA_VISIBLE_DEVICES"] = "1" 注意，服務器看到的GPU序號和手動在Linux上查看的序號是相反的，比如  那麼我們指定os.environ[「CUDA_VISIBLE_DEVICES」] = 「0」的話，你會發現在watch nvidia-smi指令下，2號GPU正在滿負荷的運行！ ---------

>>阅读原文<<