cuda設定指定的GPU可見
可設定環境變數CUDA_VISIBLE_DEVICES,指明可見的cuda裝置
方法1: 在/etc/profile或~/.bashrc的配置檔案中配置環境變數(/etc/profile影響所有使用者,~/.bashrc影響當前使用者使用的bash shell)
在/etc/profile檔案末尾新增以下行:
export CUDA_VISIBLE_DEVICES=0,1 ##僅顯示卡裝置0,1GPU可見。可用的GPU可通過nvidia-smi -L命令檢視
:wq儲存並退出
source /etc/profile使配置檔案生效
方法2:若上述配置無效,可在執行cuda程式時指明引數,如
CUDA_VISIBLE_DEVICES =0,1./cuda_executable
另附上一些參考文章:http://acceleware.com/blog/cudavisibledevices-masking-gpus
http://stackoverflow.com/questions/39649102/how-do-i-select-which-gpu-to-run-a-job-on
Environment Variable Syntax ResultsCUDA_VISIBLE_DEVICES=1
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