指定GPU运行python程序

1、命令行运行python程序时

一、首先查看哪些GPU空闲，nvidia-smi显示当前GPU使用状况。

nvidia-smi 

二、而后指定空闲的GPU运行python程序。

CUDA_VISIBLE_DEVICES=0,2,3 python test.py 

2、在python程序中指定GPU

import os os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3" 

PS：周期性输出GPU使用状况

每 10s 显示一次GPU使用状况：

watch -n 10 nvidia-smi