選擇要用的GPU: CUDA_VISIBLE_DEVICES

服務器中有多個GPU，選擇特定的GPU運行程序可在程序運行命令前使用：CUDA_VISIBLE_DEVICES=0命令。0爲服務器中的GPU編號，可以爲0, 1, 2, 3等，表明對程序可見的GPU編號。

1. 命令：

CUDA_VISIBLE_DEVICES=1     # 只有編號爲1的GPU對程序是可見的，在代碼中gpu[0]指的就是這塊兒GPU
CUDA_VISIBLE_DEVICES=0,2,3 # 只有編號爲0,2,3的GPU對程序是可見的，在代碼中gpu[0]指的是第0塊兒，
						   # gpu[1]指的是第2塊兒，gpu[2]指的是第3塊兒
CUDA_VISIBLE_DEVICES=2,0,3 # 只有編號爲0,2,3的GPU對程序是可見的，但是在代碼中gpu[0]指的是第2塊兒，
						   # gpu[1]指的是第0塊兒，gpu[2]指的是第3塊兒
--------------------- 
# 作者：__Lingyue__ 
# 來源：CSDN 
# 原文：https://blog.csdn.net/cfarmerreally/article/details/80321276 
# 版權聲明：本文爲博主原創文章，轉載請附上博文鏈接！

2. 使用

臨時設置：

Linux： export CUDA_VISIBLE_DEVICES=1
windows:  set CUDA_VISIBLE_DEVICES=1

永久設置：

linux:
在~/.bashrc 的最後加上export CUDA_VISIBLE_DEVICES=1，然後source ~/.bashrc
windows:
打開我的電腦環境變量設置的地方，直接添加就行了。