1. 程式人生 > >cuda設定指定的GPU可見

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                               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