一、命令行运行python程序时
1、首先查看哪些GPU空闲,nvidia-smi显示当前GPU使用情况。
nvidia-smi
2、然后指定空闲的GPU运行python程序。
CUDA_VISIBLE_DEVICES=0,2,3 python
二、在python程序中指定GPU
import os
["CUDA_VISIBLE_DEVICES"] = "0,2,3"
PS:周期性输出GPU使用情况
每 10s 显示一次GPU使用情况:
watch -n 10 nvidia-smi