服务器GPU查看和设置

查看GPU占用情况

nvidia-smi

设置程序运行GPU

代码中加入

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

注意： 服务器看到的GPU序号和手动在Linux上查看的序号是相反的，比如

如果指定os.environ[“CUDA_VISIBLE_DEVICES”] = “0”，则在watch nvidia-smi指令下，2号GPU正在满负荷的运行！