type
status
date
slug
tags
summary
category
password
icon
nvitop: 史上最强GPU性能实时监测工具
引言
相信大家在用NVIDIA-GPU训练网络模型的时候,都会习惯性的在终端
nvidia-smi
一下吧?最直接的目的是为了查看哪些卡正在使用,哪些卡处在空闲,然后挑选空闲的卡号进行网络训练。「了解哪块卡处在空闲只是普通算法工程师的普通需求」
咱们作为一名资深的算法工程师,毕竟身兼多职:上要开发AI算法,下要管理服务器,左要带新人,右要PPT汇报上级。
对于管理服务器:刚买的新服务器你得装系统吧?得装
Driver
,Cuda
,Cudnn
吧?时不时还得盯一下服务器各个卡的运行状况,毕竟刚入职的小年青有时候一顿操作,一个人占用全服务器95%以上的内存把服务器直接卡死也不是没有可能。nvitop
是一个非常全面的NVIDIA-GPU设备运行状况的实时监控工具,它将GPU利用率,显存占比,卡号使用者,CPU利用率,进程使用时间,命令行等等集于一身,并以差异化的颜色进行个性化展示,安装过程也非常简单,强烈大家推荐使用,让自己在管理服务器的时候事半功倍!以下图1展示了
nvitop
和nvidia-smi
命令的界面对比结果:
图1. nvitop(左) VS nvidia-smi(右)
nvitop效果展示
「NviTop」
nvitop
是一款交互式NVIDIA-GPU设备性能&资源&进程的实时监测工具。相比于
nvidia-smi
命令,nvitop
在实时监控GPU设备资源&性能上具备全方位优势:- 以更美观的颜色,和更直观的进度条实时展示某块GPU卡所处进程的GPU&CPU内存以及利用率占比
- 作为资源监控器,它包括如下功能:树状视图、环境变量查看、进程过滤、进程指标检测等
- 可追踪某个单项进程在GPU&CPU上内存和利用率占比的历史纪录,并利用Bar直观展示
- 可直观展示某块GPU的使用者、使用时间、使用命令行、GPU和CPU占用率记录
- 可使用工具提供的API搭建自定义监控工具
完整API说明文档请移步:https://nvitop.readthedocs.io
- 作者:GraphPKU
- 链接:graphpku.cn/article/16aeef0f-a2af-4279-8cf4-ebd71c66a5d1
- 声明:本文采用 CC BY-NC-SA 4.0 许可协议,转载请注明出处。