『CV学习笔记』NVIDIA GPU监控工具nvidia-smi/gpustat/nvtop/nvitop

本文介绍了四个用于监控NVIDIA GPU的工具:nvidia-smi是最有名的,gpustat提供彩色简约显示,nvtop提供详细信息但需要root权限,nvitop则结合了详细信息和便捷安装。nvitop还具备三种展示模式,可以显示每个进程的详细信息。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

NVIDIA GPU监控工具nvidia-smi/gpustat/nvtop/nvitop

一. nvidia-smi(最有名,没有之一🤚🤚)

# 直接在终端输入nvidia-smi可以得到显卡情况:
nvidia-smi
# 直接在终端输入nvidia-smi可以得到显卡情况:下面是每隔0.1秒刷新一次。
watch -n 0.1 nvidia-smi

在这里插入图片描述

二. gpustat(彩色并简约的显示🤚🤚🤚🤚)


                
03-29
### 关于 nvitop 的介绍 `nvitop` 是一款用于监控 NVIDIA GPU 使用情况的工具,它能够提供实时的 GPU 资源占用信息以及进程管理功能。该工具基于 `nvidia-ml-py3` 库开发,并提供了类似于 `htop` 的交互界面[^3]。 以下是关于安装和使用的详细介绍: --- #### 安装方法 可以通过 Python 的包管理器 `pip` 来安装 `nvitop` 工具。确保已安装最新版本的 pip 和 setuptools 后运行以下命令: ```bash pip install --upgrade nvitop ``` 如果需要支持 CUDA 进程跟踪,则还需要额外安装依赖库 `pycuda`: ```bash pip install pycuda ``` 对于某些环境可能需要手动配置 NVIDIA 驱动程序及其对应的 ML SDK 版本,请确认驱动已经正确安装并满足最低需求[^4]。 --- #### 基本使用指南 启动 `nvitop` 可通过简单的命令完成: ```bash nvitop ``` 这会打开一个图形化终端界面,在其中可以查看当前系统的 GPU 列表、每张卡上的内存分配详情、计算能力指标等数据。此外还允许用户操作相关联的任务比如终止特定 ID 所属的应用程序实例。 一些常用快捷键如下所示: - `/`: 搜索指定名称或者 PID 对应项; - `q`: 退出应用; - 方向键/Tab 键切换选中项目; 更多高级特性可通过阅读官方文档获取更多信息[^5]。 --- ### 示例脚本展示如何集成到自动化流程里 下面给出一段简单例子说明怎样利用 API 接口形式调用而非仅限 CLI 模式下工作: ```python import nvitop gpus = nvitop.Device.all() # 获取所有可用设备对象列表 for gpu in gpus: print(f'GPU {gpu.index}:') print(f' Name : {gpu.name}') print(f' Memory Total: {gpu.memory_total_human()}') print(f' Memory Used : {gpu.memory_used_human()}') processes = gpu.processes() if processes is not None: for pid, process in processes.items(): print(f' Process {pid} uses {process.used_memory_human()} memory.') ``` 上述代码片段展示了遍历检测每一枚显卡基本信息及关联线程资源消耗状况的方法[^6]。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

AI大模型前沿研究

感谢您的打赏,我会继续努力!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值