CUDA GPU 隔離 CUDA_VISIBLE_DEVICES

場景： 有一臺服務器，服務器上有多塊兒GPU能夠供使用，但此時只但願使用第2塊和第4塊GPU，可是咱們但願代碼能看到的仍然是有兩塊GPU，分別編號爲0,1，這個時候咱們能夠使用環境變量CUDA_VISIBLE_DEVICES來解決這個問題。 好比：linux CUDA_VISIBLE_DEVICES=1 只有編號爲1的GPU對程序是可見的，在代碼中gpu[0]指的就是這塊兒GPU CUDA_VIS

>>阅读原文<<