大家都知道,nvidia-smi是Nvidia显卡命令行管理套件,基于NVML库,旨在管理和监控Nvidia GPU设备。然而通常这只是显示此时的gpu使用情况,这里我给大家推荐一个小工具,可以实时的监控gpu的使用情况:

gpustat,直接pip install gpustat即可安装,gpustat基于nvidia-smi,可以提供更美观简洁的展示,结合watch命令,可以动态实时监控GPU的使用情况。安装完成后输入下面的命令:

watch --color -n1 gpustat -cpu 

即可看到实时的gpu数据:

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐