cuda设置指定的GPU可见

Environment Variable Syntax                               Results
CUDA_VISIBLE_DEVICES=1              Only device 1 will be seen

CUDA_VISIBLE_DEVICES=0,1            Devices 0 and 1 will be visible

CUDA_VISIBLE_DEVICES=”0,1”          Same as above, quotation marks are optional

CUDA_VISIBLE_DEVICES=0,2,3          Devices 0, 2, 3 will be visible; device 1 is masked

临时设置:

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

永久设置：

linux:
在~/.bashrc 的最后加上export CUDA_VISIBLE_DEVICES=1，然后source ~/.bashrc
windows:
打开我的电脑环境变量设置的地方，直接添加就行了。