GPU性能实时监控的几种软件

在深度学习服务器上，各种模型的训练，需要监控GPU的情况，并且需要根据使用状态来切换不同的GPU上。

有以下四款软件，可以很好的进行GPU状态监控。

1. nvidia-smi

一个跨平台工具，用于监控和管理NVIDIA GPU的状态和性能。它支持所有标准的NVIDIA驱动程序支持的Linux发行版以及从WindowsServer 2008 R2开始的64位系统。这个工具通常与CUDA工具包一起安装，是NVIDIA显卡驱动的一部分。

常用命令：watch -n 2 nvidia-smi

2. gpustat

一个基于nvidia-smi的命令行工具，用于监控和分析GPU（GPU）的状态和使用情况。简约显示。

安装：

# 系统包安装
sudo apt install gpustat
# 作为python库安装
pip install gpustat

运行：

watch -n 1 -c gpustat --color
# 或者
gpustat -i

3. nvtop

一个专为NVIDIA GPU设计的任务监视器，类似于htop，但提供了更直观的用户界面和更多的进程信息。

sudo apt install nvtop
# 安装完毕后直接运行
nvtop

4. nvitop（重点推荐）

一款交互式的 NVIDIA GPU 设备性能、资源和进程的实时监测工具。"nvitop" 在实时监控 GPU 设备资源和性能方面具有全方位优势，包括更美观的颜色和更直观的进度条来展示某块 GPU 卡所处进程的 GPU & CPU 内存以及利用率占比 。此外，它还支持树视图、环境变量查看、进程过滤、进程指标监控等多种功能和选项 。因此，可以推断 "nvitop" 是一个高颜值的实时监控工具，专为监控 NVIDIA GPU 设备性能和资源而设计。

官网参考：Welcome to nvitop’s documentation! — nvitop: the one-stop solution for GPU process management. documentation

安装：

pip install nvitop

运行：三种模式 auto，compact，full

nvitop -m full