I am Charmie

メモとログ

CUDAで使うGPUを環境変数で設定

ここに書いてある通り,CUDA_VISIBLE_DEVICESという環境変数にCUDAが使用するGPUのデバイス番号を設定すればいい

.bashrcに以下の命令を書き足せばOKかな?

# 0番と1番のGPUを使用可能にする場合
export CUDA_VISIBLE_DEVICES=0,1
# 0番と1番のGPUを使用可能にする場合
export CUDA_VISIBLE_DEVICES=0,1