python NVIDIA显卡查看工具 gpustat

more than nvidia-smi

Posted by Rpl on December 6, 2019

gpustat相比于NVIDIA显卡的查看工具 nvidia-smi,界面更加的美观和简洁。结合 watch 命令,可以动态实时监控 GPU 的使用情况。

1
gpustat -i

1

gpustat是python开发的一个包,可以直接使用pip安装。

值得一提的是,gpustat可以 使用 gpustat –josn 命令将gpu的所有信息通过json格式展示出来。使用脚本程序监测gpu太方便了。(详情点击: linux资源监测 – gpu)

1
gpustat --json

5

另外,由于gpustat是基于nvidia-smi的,它本质上是获取nvidia-smi的信息,然后在重新编辑/简洁化, 因此运行 nvidia-smi daemon,gpustat的查询速度会更快。

gpustat的使用非常简单,具体可以参考 gputat的pypi文档: https://pypi.org/project/gpustat/, 也可使用

1
gpustat -h

查看具体参数用法。

6

关联文章:

  1. linux资源监控——获取GPU信息