RuntimeError: CUDA out of memory. Tried to allocate 模型训练 GPU 显存不够报错总结

享受学术探讨的欢乐，传递温暖，希望能够帮助到刚刚入门的同学

• 专栏：墨理有话说：一些读研、论文写作、Bug 高效排除 方面建议
• 感谢每位读者大大、学术探讨小伙伴的支持和认可

具体报错

这个报错，遇到好多次了、今天做个记录、好记性 不如 记笔记、写博文

• 具体报错如下

RuntimeError: CUDA out of memory. Tried to allocate 1018.00 MiB (GPU 0; 7.79 GiB total capacity; 4.72 GiB already allocated; 853.50 MiB free; 1.52 GiB cached)

简单分析

• 对于代码而言、当前运行这个代码的服务器而言
• 通常遇到类似报错，的确是 GPU 显存 不够用了

这里记录常见的两种情况，有机会遇到再更新

训练时遇到

• 训练时、遇到这种报错、解决思路是，调整 batch_size ，batch_size 最小 可以设置 为 1
• 示例如下：

• 考虑一下，是不是你当前 测试 送入 模型的 图片太大了
• 如果你确实是 GPU 太小、可以考虑换个服务器 显卡

测试时遇到

• 建议首先考虑一下，是不是你当前 测试 送入 模型的 图片太大了
• 这个情况，遇到好多次了，示例如下

2022年的 某一次

• 这个同学是，超分代码测试，部分图片顺利生成、到了 某几个 图片 ，就 报 Cuda 显存不够的问题

第 N 次 遇到 类似讨论，还是其他 同学 提醒指正 ，因此 觉得有必要 写这个博文

---

博主 AI 领域八大干货专栏、诚不我欺

---

• 计算机视觉： Yolo专栏、一文读懂
• 计算机视觉：图像风格转换–论文–代码测试
• 计算机视觉：图像修复-代码环境搭建-知识总结
• 计算机视觉：超分重建-代码环境搭建-知识总结
• 深度学习：环境搭建，一文读懂
• 深度学习：趣学深度学习
• 落地部署应用：模型部署之转换-加速-封装
• CV 和 语音数据集：数据集整理

---

预祝各位 前途似锦、可摘星辰

---

• 作为全网 AI 领域 干货最多的博主之一，❤️ 不负光阴不负卿 ❤️
• ❤️ 过去的每一天、想必你也都有努力、祝你披荆斩棘、未来可期

• 深度学习模型训练推理——基础环境搭建推荐博文查阅顺序【基础安装—认真帮大家整理了】
• 计算机视觉：硕博士，交流、敬请查阅
• 点赞 收藏 ⭐留言 都是博主坚持写作、更新高质量博文的最大动力！