服务器GPU查看和设置

查看GPU占用情况

nvidia-smi

-周期性的输出显卡的使用情况：

watch -n 10 nvidia-smi

命令行参数-n后边跟的是执行命令的周期，以s为单位。

设置程序运行GPU

1.代码中加入

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

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

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

2.命令行CUDA_VISIBLE_DIVICES=1 python xxx.py