通过修改文件参数,让你的gpustat更加炫酷

通过修改文件参数,让你的gpustat更加炫酷

gpustat

gpustat是一款监控 GPU 的使用情况的工具,可以让你轻松监控GPU状态。首先你需要通过命令安装

pip install gpustat

然后在任意目录内输入gpustat即可调用这个工具

修改文件参数

初始安装的gpustat工具只能显示当前时刻的GPU状态,需要通过gpustat -i等附加参数实现监控,为了方便起见,我们还需要进一步对gpustat进行设置。

我们需要修改 环境/lib/python3.6/site-packages/gpustat 里的__main__.py文件

如果你不知道你的环境路径,可以再次pip install gpustat一次,查看gpustat工具的安装路径。

修改刷新率

打开__main__.py文件后,我们需要对94行(随版本不同可能有出入)的default的赋值进行修改

    parser.add_argument(
        '-i', '--interval', '--watch', nargs='?', type=float, default=0.1,
        help='Use watch mode if given; seconds to wait between updates'

这个参数代表的是GPU监控的刷新率,安装时默认为0。当这个值为0的时候,系统只print一次GPU状态,我们可以通过把这个值改为0.1(即每0.1s刷新一次)实现实时监控。当然你也可以改成任何你希望的刷新率。
请注意,笔者使用ssh进行长时间刷新有概率导致窗口卡死,原因未知。如果你遇到这种情况,可以使用top查找名为gpustat的PID,然后 sudo kill -9 PID。 不过仍建议避免长期使用这个方法进行监控

调出功率显示

虽然这个功能可能大部分时间并不会对你的工作产生实质性帮助,但是看着爽
还是__main__.py文件,90行

    parser.add_argument(
        '-P', '--show-power', nargs='?', const='draw,limit',
        choices=['', 
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值