Repolode
提交项目
wookayin

wookayin/gpustat

一个命令行工具，以简洁易读的格式查看 NVIDIA、AMD 和华为昇腾 GPU 的使用率、显存、温度和进程信息。

可替代:nvidia-smi

在 GitHub 查看官网
Star
4,387
Fork
287
未关闭 Issue
27
最近提交
上个月

这是什么？

gpustat 是一个用 Python 编写的小型命令行工具，用于查询和监控 GPU 状态。每块 GPU 输出一行，包含型号、温度、使用率、显存占用以及正在运行的进程，比 nvidia-smi 的输出更紧凑。

工具支持通过 NVML 读取 NVIDIA GPU，通过 ROCm SMI 读取 AMD GPU，并通过 npu-smi 读取华为昇腾 910B NPU。后端可自动识别，也可用参数指定。可选显示用户、PID、完整命令、风扇转速、编解码器占用和功耗等信息，并支持按固定间隔刷新的 watch 模式。

除命令行外，它还提供 Python API 供代码内调用，并可输出 JSON，便于脚本处理。

核心功能

  • 01每块 GPU 一行，显示型号、温度、使用率和显存占用。
  • 02列出每块 GPU 上的进程，包括所属用户、PID 和显存占用。
  • 03支持 NVIDIA（NVML）、AMD（ROCm SMI）和华为昇腾 910B 后端。
  • 04提供 watch 模式，可按指定间隔自动刷新。
  • 05支持 JSON 输出和 Python API，方便脚本集成。

适用场景

  • 在共享工作站或服务器上启动训练任务前，查看哪些 GPU 空闲。
  • 训练长时间运行的模型时，持续观察显存和使用率。
  • 在监控脚本中以 JSON 形式采集 GPU 指标。
  • 通过 SSH 快速查看 GPU 状态。

快速上手

  1. 1

    从 PyPI 安装 gpustat

    pip install gpustat
  2. 2

    如果使用 NVIDIA GPU，请安装 NVIDIA 官方 NVML 绑定（而非 pynvml）

    pip install nvidia-ml-py
  3. 3

    运行 gpustat 查看 GPU 状态

    gpustat

使用前须知

  • 对于 NVIDIA GPU，需要 NVIDIA 驱动（gpustat 1.0+ 要求 450.00 或更高版本）。
  • 使用 pynvml 包而非 nvidia-ml-py 将无法正常工作。

收录于专题

GPU 加速的 AI 训练与推理工具

标签

#gpu#nvidia#amd#monitoring#command-line#nvidia-smi#deep-learning#gpustat#python

常见问题

gpustat 是什么？
一个命令行工具，以简洁易读的格式查看 NVIDIA、AMD 和华为昇腾 GPU 的使用率、显存、温度和进程信息。 gpustat 是一个用 Python 编写的小型命令行工具，用于查询和监控 GPU 状态。每块 GPU 输出一行，包含型号、温度、使用率、显存占用以及正在运行的进程，比 nvidia-smi 的输出更紧凑。
gpustat 可以免费使用吗？用的什么许可证？
gpustat 以 MIT 许可证开源。用于商业产品前，请先确认许可证条款。
gpustat 还在维护吗？
半年内有更新。gpustat 最近一次提交是在 2026年10月8日。
gpustat 适合用来做什么？
在共享工作站或服务器上启动训练任务前，查看哪些 GPU 空闲。 训练长时间运行的模型时，持续观察显存和使用率。 在监控脚本中以 JSON 形式采集 GPU 指标。 通过 SSH 快速查看 GPU 状态。
gpustat 怎么安装和运行？
从 PyPI 安装 gpustat: pip install gpustat 如果使用 NVIDIA GPU，请安装 NVIDIA 官方 NVML 绑定（而非 pynvml）: pip install nvidia-ml-py 运行 gpustat 查看 GPU 状态: gpustat
在哪里获取 gpustat？
源代码和安装说明在 GitHub 上：https://github.com/wookayin/gpustat

概要由 AI 根据项目 README 生成；GitHub 数据每日更新。

相关项目