CUDA GPU隔離CUDA_VISIBLE_DEVICES

シーン:
1台のサーバーがあって、サーバーの上で多くのGPUが使用することができて、しかしこの時ただ第2のブロックと第4のブロックのGPUを使うことを望んで、しかし私達はコードが見えることができることを望んで依然として2つのGPUがあって、それぞれ番号は0,1で、この時私達は環境変数CUDA_を使うことができますVISIBLE_DEVICESはこの問題を解決します.例:

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  

環境変数の設定方法
一時的な設定:

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

永続的な設定:

linux:
 ~/.bashrc      export CUDA_VISIBLE_DEVICES=1，  source ~/.bashrc
windows:
               ，       。