gpustat 一个方便查询 GPU 状态的命令行程序

如果你正在参与深度学习的研究或工作，那么必不可少的会需要查看GPU的状态，去分析数据，而GpuStat就是Python开发的一个方便查询GPU状态的一个命令行程序。

注意：该功能仅适用于NVIDIA图形设备，目前不支持AMD。

展示效果

安装指南

通过 PyPI安装:

pip install gpustat

如果没有系统用户ROOT权限，那么建议在用户空间下安装:

pip install --user gpustat.

可以通过Github下载最新的版本使用:

pip install git+https://github.com/wookayin/gpustat.git@master

注意：v1.0版本的gpustat 仅支持 Python 3.4+. 老版本的 (python 2.7, <3.4), y可以继续使用gpustat v0.x版本。

使用

$ gpustat

Options:

• --color : 强制彩色输出
• --no-color : 禁止彩色输出
• -u, --show-user :显示进程所有者的用户名
• -c, --show-cmd : 显示进程名称
• -p, --show-pid : 显示进程的PID
• -F, --show-fan : 显示GPU风扇速度
• -P, --show-power : 显示GPU的功耗和/或极限(draw or draw,limit)
• --watch, -i, --interval : 若指定，则以监视模式运行（相当于watch gpustat)
• --json :JSON输出（测试功能）

小技巧

• 定期观察, 可以使用

  gpustat --watch
  

  或者

  gpustat -i
  

  老版本的用户可以使用 watch --color -n1.0 gpustat --color

• 运行nvidia-smi daemon (需要ROOT权限) 可以查询更快，更节约cpu资源

更多学习使用技巧，可以[参照官网