开发者社区> 问答> 正文

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

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

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

展示效果

img

安装指南

通过 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资源

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

展开
收起
huc_逆天 2020-05-21 16:23:56 3903 0
1 条回答
写回答
取消 提交回答
  • 技术架构师 阿里云开发者社区技术专家博主 CSDN签约专栏技术博主 掘金签约技术博主 云安全联盟专家 众多开源代码库Commiter

    欢迎大家分享哦~

    2020-05-21 16:53:51
    赞同 展开评论 打赏
问答排行榜
最热
最新

相关电子书

更多
DeepStream: GPU加速海量视频数据智能处理 立即下载
阿里巴巴高性能GPU架构与应用 立即下载
GPU在超大规模深度学习中的发展和应用 立即下载