Jupyter Notebook 运行内存溢出怎么办

风瑶姑娘_8596

风瑶姑娘_8596

2026-08-08

1005人浏览

原创

内存溢出不是代码错误,而是资源超限;jupyter不自动释放内存,pytorch显存缓存、pandas全量加载、大文件读取等场景易引发oom,需分块处理、及时清理引用并监控内存指标。

jupyter notebook 运行内存溢出怎么办

内存溢出不是“代码写错了”,而是资源使用超出了当前内核或系统能承受的边界。Jupyter 本身不自动释放内存,加上 Python 的引用计数和垃圾回收机制在交互式环境中容易滞后,问题会集中爆发在 DataFrame 加载、PyTorch 训练、大文件读取等场景。

为什么 pd.read_csv() 一执行就报 MemoryError

因为默认行为是把整个文件一次性加载进内存。哪怕只是 2GB 的 CSV,read_csv() 也会尝试分配连续内存块,而你的内核可能只剩 1.5GB 可用空间。

  • 用 chunksize 参数分块读取:每次只加载一部分,处理完立刻丢弃引用
  • 避免写 df = pd.read_csv('big.csv') 后直接做 df.groupby(...).apply(...) —— 这会让中间结果全留在内存里
  • 读取时指定列(usecols)和数据类型(dtype),比如把 int64 换成 int32 或 category,可省下 30%~70% 内存

PyTorch 在 Jupyter 里跑几轮就 Kernel died

显存没释放,不是内存没释放。CUDA 缓存分配器会把显存“预留”着供下次复用,但不会还给系统。你看到 torch.cuda.memory_reserved() 持续上涨,就是它在悄悄囤货。

Jupyter Notebook 7.5.4
Jupyter Notebook 7.5.4

Jupyter Notebook 7.5.4 官方 PyPI wheel 包下载,适合用于本地交互式代码运行、数据分析、教学演示和配置文件实践。

下载
  • 训练循环里必须加 optimizer.zero_grad() 和 loss.backward() 后的 del loss, output —— 不然计算图一直被引用
  • 别把带 requires_grad=True 的张量塞进列表:losses.append(loss) 是高危操作;改用 losses.append(loss.item())
  • 手动清缓存:torch.cuda.empty_cache() 可以强制释放未被引用的显存,但不能解决根本泄漏

运行完代码,内存还是占着不放

Jupyter 内核默认保持活跃状态,所有变量、对象、导入模块都留在内存里,这是设计使然,不是 bug。你关掉浏览器标签页,内核还在后台跑着。

  • 点击菜单栏 Kernel → Restart & Clear Output,比单纯 Restart 更彻底
  • 命令行里查内核进程:jupyter kernel list,再用 kill -9 PID 强制终止残留内核(尤其当你改过内核配置后)
  • 长期运行任务建议改用脚本 + python train.py,而不是在 Notebook 里反复 Shift+Enter

服务器级内存不足导致 Internal Server Error

这不是 Notebook 本身的问题,是 Jupyter Server 进程被系统 OOM Killer 杀掉了。常见于云服务器上多个用户共用一台机器,或你开了太多内核没关。

  • 检查配置文件 ~/.jupyter/jupyter_server_config.py,设 c.ServerApp.max_kernels = 3 防止失控创建
  • 启动时限制总内存:jupyter notebook --ServerApp.memory_limit=6g,让内核在超限时主动退出,而非被强杀
  • 配合 free -h 和 ps aux --sort=-%mem | head -10 定期看谁在偷偷吃内存,特别是 python 或 node 进程

最常被忽略的一点:内存问题往往不是单点故障,而是多个小泄漏叠加的结果——一个没删的 DataFrame、一次没 .item() 的 loss、一个 chunk 读取后忘了 del chunk,三者加起来就能压垮内核。盯住 torch.cuda.memory_allocated() 和 gc.get_count(),比等崩溃后再排查更高效。

相关文章

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

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

下载

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

相关专题

更多
LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

2026.09.30

60

10

LLVM RISC-V参数配置教程
LLVM RISC-V参数配置教程

本专题介绍LLVM对RISC-V基础ISA和扩展的支持方式,涵盖RV32、RV64、标准扩展、实验性扩展、厂商扩展、-menable-experimental-extensions和版本差异。

2026.09.30

40

14

LLVM IR中间表示入门指南
LLVM IR中间表示入门指南

本专题整理LLVM IR的核心概念,包括中间表示作用、模块结构、函数、基本块、SSA形式、类型系统和常见语法,帮助新手理解LLVM编译流程中的关键层。

2026.09.30

40

12

PDF转图片方法
PDF转图片方法

需要把 PDF 页面用于上传、预览、分享或图片归档时,PDF 转图片方法专题整理 JPG/PNG 格式选择、逐页导出、清晰度设置、批量下载和结果检查等流程,帮助用户稳定完成 PDF 图片化处理。

2026.09.30

40

26

PixTV AI视频生成与无限画布创作
PixTV AI视频生成与无限画布创作

PixTV专题整理AI视频与视觉内容创作相关功能使用教程,涵盖AI生图、视频生成、无限画布、多模型创作、素材管理、声音音乐及视频剪辑等功能,帮助用户快速掌握PixTV从创意到成片的完整制作方法。

2026.09.29

40

15

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

2026.09.23

240

15

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

2026.09.23

160

15

Buffalo框架零基础入门教程
Buffalo框架零基础入门教程

本专题整理Buffalo框架入门内容,涵盖Go环境准备、buffalo CLI安装、新项目生成、目录结构说明、dev热加载启动、数据库连接配置与常见报错排查,帮助新手按约定优于配置的思路跑通第一个Buffalo框架应用。

2026.09.23

120

15

Conan创建软件包配方指南
Conan创建软件包配方指南

本专题介绍通过conanfile.py创建软件包的方法,讲解包名、版本、依赖和构建设置等基础信息,以及source、build、package、package_info等常用方法的作用及编写思路。

2026.09.22

80

12

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程