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=['',