使用两种方法,实时查看 GPU 使用情况
1. nvidia-smi
使用终端命令 nvidia-smi 查看显卡信息
如果你想实时检测显卡使用情况,添加 watch -n 即可
watch -n 4 nvidia-smi
其中,4 是指 4 秒刷新一次终端,可以根据自己的需求设置
2.gpustat
安装过程很简单,直接 pip 即可(本人是这样)
使用 gpustat --json 以 json 形式呈现 gpu 信息
使用 gpustat -i 命令可以查看用户使用 gpu 情况
使用以下命令,可以查看更具体的信息,比如用户 xxx 的 xxx 进程占用情况
watch --color -n 1 gpustat -cpu