在GPU上跑代码

网上查的是，首先查看自己的GPU

nvidia-smi

然后在python文件中指定GPU：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

但是我试了一下一段比较费时间的程序，怎么没显示GPU在运行。