动态查看GPU显存

在Linux中可以使用watch -n 0.1 nvidia-smi命令来动态查看GPU中的显存消耗情况

watch命令的用法可以看这里

### 如何检查 GPU 显存使用情况 通过 `pynvml` 模块可以实现对 NVIDIA GPU显存使用情况进行监控。以下是具体方法: #### 使用 pynvml 查看 GPU 显存使用情况 可以通过安装并导入 Python 库 `pynvml` 来获取详细的 GPU显存信息。以下是一个完整的代码示例,用于展示当前系统的 GPU 显存使用状况[^1]。 ```python import pynvml def get_gpu_memory_usage(): # 初始化 NVML pynvml.nvmlInit() device_count = pynvml.nvmlDeviceGetCount() # 获取设备数量 gpu_info_list = [] for i in range(device_count): handle = pynvml.nvmlDeviceGetHandleByIndex(i) # 获取指定索引的 GPU 句柄 memory_info = pynvml.nvmlDeviceGetMemoryInfo(handle) # 获取内存信息 total_memory_mb = memory_info.total / (1024 ** 2) # 总显存大小(MB) used_memory_mb = memory_info.used / (1024 ** 2) # 已使用的显存大小(MB) gpu_name = pynvml.nvmlDeviceGetName(handle).decode('utf-8') # 获取 GPU 名称 info_dict = { 'index': i, 'name': gpu_name, 'total_memory_MB': round(total_memory_mb, 2), 'used_memory_MB': round(used_memory_mb, 2) } gpu_info_list.append(info_dict) pynvml.nvmlShutdown() # 关闭 NVML return gpu_info_list if __name__ == "__main__": result = get_gpu_memory_usage() for item in result: print(f"GPU {item['index']}: Name={item['name']}, Total Memory={item['total_memory_MB']} MB, Used Memory={item['used_memory_MB']} MB") ``` 上述脚本会返回每张 GPU 卡的信息,包括总显存量、已用显存量以及对应的名称。 #### 命令行工具 nvidia-smi 除了编程方式外,还可以直接在 Linux 终端中运行命令来快速查看 GPU 显存的状态。例如,执行如下命令即可获得更简洁的结果[^3]: ```bash nvidia-smi --query-gpu=memory.free,memory.used,gpu_utilization --format=csv ``` 此命令将显示各 GPU 的可用显存、已用显存GPU 利用率数据。 ---
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值