まずは刺さっているGPUを確認する
$ nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 387.26 Driver Version: 387.26 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 108... Off | 00000000:01:00.0 Off | N/A |
| 0% 37C P8 21W / 275W | 10MiB / 11158MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
| 1 GeForce GTX 108... Off | 00000000:06:00.0 Off | N/A |
| 0% 28C P8 9W / 275W | 10MiB / 11172MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
+-----------------------------------------------------------------------------+
GPU0番だけ使用
export CUDA_VISIBLE_DEVICES="0"
GPU1番だけ使用
export CUDA_VISIBLE_DEVICES="1"
全GPUを使用
export CUDA_VISIBLE_DEVICES="0,1"
参考文献
複数GPUを活用するヒント | 九州大学情報基盤研究開発センター