wookayin/gpustat

解决的问题

gpustat 提供了一个比 nvidia-smi 更简洁且易读的替代方案，用于监控 NVIDIA GPU 状态。它简化了 GPU 指标（如温度、利用率和内存使用情况）的呈现，使用户能够更轻松地追踪哪些进程和用户正在占用 GPU 资源。

运作原理

它作为 NVIDIA 官方 Python 绑定库 nvidia-ml-py (NVML) 的封装工具。它可以作为命令行工具运行，也可以作为 Python 库导入，以编程方式查询 GPU 统计数据。

适用对象

使用 NVIDIA GPU 的开发人员和研究人员，需要一种快速且清晰的方式来实时监控硬件利用率和进程所有权。

特色亮点

• 全面指标：显示 GPU 索引、名称、温度、利用率和内存使用情况。
• 进程追踪：可显示进程所有者的用户名、进程名称和 PID。
• 灵活输出：支持彩色终端输出、用于自动化的 JSON 格式，以及用于定期更新的监控模式。
• Python API：允许集成到其他 Python 项目中，通过代码查询 GPU 统计数据。