大家都知道,nvidia-smi
是Nvidia显卡命令行管理套件,基于NVML库,旨在管理和监控Nvidia GPU设备。然而通常这只是显示此时的gpu使用情况,这里我给大家推荐一个小工具,可以实时的监控gpu的使用情况:
gpustat
,直接pip install gpustat
即可安装,gpustat基于nvidia-smi
,可以提供更美观简洁的展示,结合watch命令,可以动态实时监控GPU的使用情况。安装完成后输入下面的命令:
watch --color -n1 gpustat -cpu
即可看到实时的gpu数据: