Linux怎么使用nvidia-smi排查深度学习显存溢出

梦敏吖_9544

梦敏吖_9544

2026-11-03

370人浏览

原创

显存溢出本质是CUDA分配器拒绝分配,非nvidia-smi显示的物理占用;关键征兆包括:GPU-Util低但Memory-Usage高、nvidia-smi进程列表为空却显存未释放、torch.cuda.memory_allocated与memory_reserved差值超500MB。

linux怎么使用nvidia-smi排查深度学习显存溢出

nvidia-smi 本身不报“显存溢出”错误,它只反映当前状态;真正触发 OOM 的是 CUDA 运行时或框架(如 PyTorch)在 cudaMalloc 失败时抛出异常。排查关键不是等报错,而是提前识别“显存快撑爆”的征兆。

怎么看显存是不是真被占满了

别只盯着 nvidia-smi 输出顶部的 Memory-Usage 行——它显示的是“已分配但未必正在用”的显存总量。真正危险的信号是:

  • GPU-Util 很低(比如 Memory-Usage 持续 >90%,说明显存被大张量占着不动,没释放
  • 运行 nvidia-smi -q -d MEMORY,对比 Used GPU Memory 和进程列表里所有 Used GPU Memory 的总和:差值 >100MB 就大概率有泄漏
  • 执行新任务时报 CUDA out of memory,但 nvidia-smi 显示“Processes”栏为空——这是典型残留句柄问题

为什么 kill 进程后显存还不释放

因为 nvidia-smi 只显示通过 CUDA API 正常注册的进程。以下情况会导致显存卡住不放:

  • Python 进程被 kill -9 或 Ctrl+C 中断,子进程(如 DataLoader 的 worker)还在持有 /dev/nvidia0 句柄
  • PyTorch 的 torch.cuda.empty_cache() 只清空缓存池,不释放已分配给 tensor 的显存;tensor 仍存活就无法回收
  • Jupyter notebook 内核崩溃后未完全退出,后台 Python 进程仍在

这时必须用 fuser -v /dev/nvidia* 查真实句柄持有者,再 kill -9 对应 PID。

Linux Security Scanner
Linux Security Scanner

Linux 安全审计工具,检查 SSH 配置、开放/监听端口、防火墙规则(ufw/iptables/nftables)、失败登录尝试、sudoers 权限...

下载

PyTorch 场景下最该检查的三件事

深度学习训练中最容易误判为“显存溢出”的,其实是框架层资源管理问题:

  • 确认是否启用了 pin_memory=True 且 num_workers > 0:worker 进程会在 GPU 上预分配显存缓冲区,即使模型没跑也会占几十 MB
  • 检查 torch.utils.data.DataLoader 是否用了 drop_last=False + batch_size 不整除数据集:最后一个 batch 可能因 shape 不一致导致隐式重复加载
  • 验证 model.to('cuda') 和 data.to('cuda') 是否成对出现:漏掉 data 移动,模型会把中间结果 fallback 到 CPU,后续操作可能触发意外的 GPU 张量创建

别忽略 nvidia-smi 的 Perf 和 Pwr 字段

当显存显示快满、但训练速度极慢时,Perf 字段常暴露真相:

  • 如果 Perf 是 P8 或 P12(而非 P0),说明 GPU 被降频了——可能是温度过高(看 Temp)或功耗封顶(看 Pwr:Usage/Cap 接近 100%)
  • 此时显存“看起来”不够,其实是计算单元跑不动,batch size 被迫调小,间接加剧显存压力
  • 用 nvidia-smi -pl 临时提功耗上限(需 root),或加装散热/清理风扇,比盲目调小 batch 更治本

显存是否真的溢出,取决于你看到的是“物理显存耗尽”,还是“CUDA 分配器拒绝再分”——后者往往藏在 /dev/nvidia* 句柄、PyTorch 的 tensor 生命周期、以及 GPU 自身的功耗策略里。盯紧 fuser 和 nvidia-smi -q 的差值,比反复重启 Python 进程更有效。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
磁盘配额是什么
磁盘配额是什么

磁盘配额是计算机中指定磁盘的储存限制,就是管理员可以为用户所能使用的磁盘空间进行配额限制,每一用户只能使用最大配额范围内的磁盘空间。php中文网为大家提供各种磁盘配额相关的内容,教程,供大家免费下载安装。

2023.06.21

4370

4

如何安装LINUX
如何安装LINUX

本站专题提供如何安装LINUX的相关教程文章,还有相关的下载、课程,大家可以免费体验。

2023.06.29

3693

10

linux find
linux find

find是linux命令,它将档案系统内符合 expression 的档案列出来。可以指要档案的名称、类别、时间、大小、权限等不同资讯的组合,只有完全相符的才会被列出来。find根据下列规则判断 path 和 expression,在命令列上第一个 - ( ) , ! 之前的部分为 path,之后的是 expression。还有指DOS 命令 find,Excel 函数 find等。本站专题提供linux find相关教程文章,还有相关

2023.06.30

2953

5

linux修改文件名
linux修改文件名

本专题为大家提供linux修改文件名相关的文章,这些文章可以帮助用户快速轻松地完成文件名的修改工作,大家可以免费体验。

2023.07.05

7290

7

linux系统安装教程
linux系统安装教程

linux系统是一种可以免费使用,自由传播,多用户、多任务、多线程、多CPU的操作系统。本专题提供linux系统安装教程相关的文章,大家可以免费体验。

2023.07.06

6922

7

linux查看文件夹大小
linux查看文件夹大小

Linux是一种自由和开放源码的类Unix操作系统,存在着许多不同的Linux版本,但它们都使用了Linux内核。Linux可安装在各种计算机硬件设备中,比如手机、平板电脑、路由器、视频游戏控制台、台式计算机、大型机和超级计算机。linux怎么查看文件夹大小呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.07.20

3953

8

linux查看ip命令
linux查看ip命令

本专题为大家提供linux查看ip命令相关文章内容,感兴趣的朋友可以免费下载体验试试。

2023.07.20

3445

4

linux查看cpu使用率
linux查看cpu使用率

在linux的系统维护中,可能需要经常查看cpu使用率,分析系统整体的运行情况。本专题为大家带来了linux查看cpu使用率的相关文章,感兴趣的朋友千万不要错过了。

2023.07.25

3679

6

linux查找文件命令
linux查找文件命令

Linux是一种自由和开放源码的类Unix操作系统,存在着许多不同的Linux版本,但它们都使用了Linux内核。Linux可安装在各种计算机硬件设备中,比如手机、平板电脑、路由器、视频游戏控制台、台式计算机、大型机和超级计算机。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.07.31

297

3

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

Buffalo框架快速入门指南
Buffalo框架快速入门指南

共0课时 | 0人学习

Valgrind支持平台说明
Valgrind支持平台说明

共0课时 | 0人学习