查看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