1. 程式人生 > >如何指定程式在某個GPU上執行

如何指定程式在某個GPU上執行

只需要新增如下兩行程式碼:

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

注意,伺服器看到的GPU序號和手動在Linux上檢視的序號是相反的,比如
這裡寫圖片描述

那麼我們指定os.environ[“CUDA_VISIBLE_DEVICES”] = “0”的話,你會發現在watch nvidia-smi指令下,2號GPU正在滿負荷的執行!