YOLOv4训练提示 Out of memory!

最新推荐文章于 2024-06-30 23:24:39 发布
最新推荐文章于 2024-06-30 23:24:39 发布
阅读量1.1k 收藏
点赞数 1
分类专栏： YOLOv4 文章标签： 深度学习 神经网络
转载注明出处。
本文链接：https://blog.csdn.net/weixin_40557160/article/details/108961229
版权
本文提供了一种应对YOLOv4训练时Out of memory问题的方法，建议降低cfg文件中的height和weight设置，如从416减小到960，以避免参数过多导致的计算资源消耗过大。核心在于理解高分辨率图片对模型计算的挑战和适当调整策略。
摘要由CSDN通过智能技术生成

YOLOv4训练提示 Out of memory!

网上的办法如果都解决不了，试试我的这个。
训练时候修改你自己的cfg文件中height和weight，往小改，官方为416，我改成960报以上错误。
height=960
weight=960

根本原因是，输入网络的图片尺寸越大，卷积核计算出最后的tensor的维度会过高，可以理解为参数爆炸，所以YOLO也不能处理过高分辨率的图片。

愿望是当打工人
关注
专栏目录
yolo v2 训练自己数据集遇到的问题
jozeeh的博客
11-27 4259
1.CUDA Error: out of memory darknet: ./src/cuda.c:36: check_error: Assertio `0' failed. 需要修改所使用的模型cfg文件中的subdivision的参数。 由subdivisions=8改成subdivisions=64。 subdivision：这个参数很有意思的，它会让你的每一个ba
自动驾驶－YOLOV5网络训练（可能会遇到五个坑）
harryxia2014的博客
02-09 3850
1、环境： 操作系统：Ubuntu20.04 Python3.9 Pytorch1.10.2 IDE：（本人使用的是ECLIPSE） 显卡：NVIDIA GeForce GTX 1060 6GB 2、网络训练 在YOLOV5目标下，选择：train.py，执行 3、可能会遇到的坑一 报：/bin/sh: pip: command not found 遇到这错误，说明环境不协调，需要安装配套的环境，具体每个人的环境不一样，需要细心解决，找资料。 4、可能会遇到的坑二 报：wandb.e
Yolo4 CUDA Error: out of memory
0x12的专栏
06-23 1614
CUDA status Error: file: ./src/dark_cuda.c : () : line: 373 : build time: May 6 2020 - 11:56:13 CUDA Error: out of memory CUDA Error: out of memory: Cannot allocate memory detector: ./src/utils.c:325: error: Assertion `0' failed. https://github.com/Ale.
深度学习CUDA Out of Memory原因总结和方法
最新发布
PeterClerk的博客
06-30 1150
CUDA Out of Memory原因总结和方法
运行YOLOv3报错： CUDA Error: out of memory
weixin_44401286的博客
11-05 1638
报错现象： 19 conv 128 1 x 1 / 1 76 x 76 x 256 -> 76 x 76 x 128 0.379 BFLOPs 20 CUDA Error: out of memory darknet: ./src/cuda.c:36: check_error: Assertion `0' failed. 已放弃 (核心已转储) 解决办法： 这是由于显存不足导致的报错，打开yolov3.cfg文件，修改subdivisions，显存不足时，
使用YOLO网络对图像进行目标检测时，CUDA Error: out of memory
Uranus的学习日记
07-16 1435
环境：Ubuntu14，CUDA8.0、Cudnn5.1、opencv3.1.0、 显卡：GTX1050 运行内存：2G 在darknet编译好后（GPU=1，CUDNN=1，OPENCV=1），下载好weights文件，试运行 ./darknet detect cfg/yolov3.cfg models/yolov3.weights data/dog.jpg 报错： CUDA Err...
YOLOv3 CUDA Error: out of memory darknet: ./src/cuda.c:36: check_error: Assertion `0‘ failed.
nicole3375的博客
07-26 300
Batch Size（批量大小）：批量大小是指在每次模型训练中同时处理的输入样本数量。Subdivisions（子分区）：子分区是将每个批量进一步划分为更小的部分进行处理的方式。通过使用子分区，可以减少 GPU 内存的需求，特别是当批量大小较大时。把subdivisions=8改成subdivisions=16，64，最好都是2的指数，并且要保证batch是subdivisions的整数倍。方法二：修改尾行 random 值 （默认值为1），此处修改为 ：random=0 （注：yolov3共有三处）
PyTorch-YOLOv3训练自己的数据集–之–排坑指南
12-21
3. **`Out of memory tried to allocate xxxx bytes`**： 这是GPU内存不足的提示。如果遇到这样的问题，可以考虑降低`train.py`中的`batch_size`，减小每次训练的样本数量，以减少对GPU内存的需求。 4. **`OSError...
yolov5torch.cuda.OutOfMemoryError: CUDA out of memory.
08-31
- *2* [训练yolov5时出现RuntimeError: CUDA out of memory](https://blog.csdn.net/weixin_43945848/article/details/126266421)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_...
利用yolov5训练自己的数据集； yolov5的安装与使用 ； yolov5源码解读
m0_67401660的博客
06-21 2697
*免责声明: 1此方法仅提供参考 2搬了其他博主的操作方法,以贴上路径. 3*1:基本命令 查看Anaconda的版本信息 查看python版本信息 打开Jupyter Notebook命令 或者 退出python输入环境: 命令行终止正在运行的程序命令 : 查看 opencv版本信息: 2:创建使用自己的虚拟环境生成一个名叫 jiance的环境,用来进行做识别任务: 进入这个环境,也就是激活这个环境 windows下: 接下来就是 在这个环境中可以下载你所需要的包 或者是退出这个环境 :查看
yolov5环境配置及训练coco128数据集
qq_57347782的博客
10-29 5379
yolov5搭建，训练coco128
基于Windows系统进行Yolov3训练模型遇到：“CUDA Error: out of memory”的解决方案
HeyLoong的博客
07-08 1527
遇到问题：CUDAError:outofmemory 原因：爆显存 Hhhhhha ！！！ 解决方案：更改训练所使用的CFG(.cfg)文件参数 （本人使用的是：yolov3-voc.cfg） （1）修改 batch subdivisions 值 （batch可较默认值64略小，subdivisions可 较默认值48略大），此处设置：batch=48， subdivisions=48 （2）修改尾行 random 值 （默认值为1），此处修改为 ：r...
在yolo训练的时候又去测试就会报错：cuda error: out of memory
热门推荐
walle的博客
08-31 2万+
在yolo训练的时候又去测试就会报错：cuda error: out of memory,  cuda.c  Assertion '0' failed.
yolov3的GUI界面（3）--解决out of memory问题，新增摄像头检测功能
perfectdisaster的博客
07-18 807
yolov3的GUI界面（3）--解决out of memory问题，新增摄像头检测功能
RuntimeError: CUDA out of memory怎么办？求助，救急
laidone的博客
07-03 369
我的代码是特别简单的一个学习网络，按理说不应该有GPU不够的问题，但是结果却是RuntimeError: CUDA out of memory，真不知道该怎么办，特别急，求帮助
out of memory--神经网络训练--case
dawnminghuang的专栏
10-23 3382
之前的工作： 由于下载到的数据集包含62个文件夹，每个文件夹有1016张图片，0-9，A-Z,a-z 。上周研究的是印刷体数字的识别，只有用的10个文件夹，这时读取的图片数目可以很大，得到的识别效果也十分好，接近98%。 现在的工作： 将处理的的范围扩大到大小写字母，这时读取的图片量剧增，以每个文件夹读取300张算，总体的图片张数是18600，在训练的时候发生了错误out of memory
运行yolov5训练数据集，报错：RuntimeError: CUDA error: out of memory CUDA kernel errors might be asynchronously
qq_44421796的博客
12-17 951
batch_size默认是16，然后我运行就显示这个错误，我查看网上的资料很多，对于我来说，我将batch_size改成4就可以成功运行了。
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0； 1.96 GiB total capacity； 1.40
weixin_45955386的博客
11-26 2122
跑yolov4测试demo报错欢迎使用Markdown编辑器新的改变功能快捷键合理的创建标题，有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能，丰富你的文章UML 图表FLowchart流程图导出与导入导出导入 with torch.no_grad(): 欢迎使用Markdown编辑器 RuntimeError: CUDA
跑yolov4模型时，显存明明足够却报错说超出显存
WhiteGive__的博客
07-08 1727
原本num_workers=4改成num_workers=2后可以跑。这个问题是因为理论上windows只能单线程nun_workers应该=0，但有的时候也能多线程，num_workers是玄学问题？在解决任务管理器gpu项目中没有cuda这个选项时，重启两三遍之后num_workers=4也可以跑了 ...
yolov5 cuda out of memory
03-16
yolov5 cuda out of memory是指在使用yolov5模型进行计算时，显存不足导致程序无法继续运行。解决方法可以尝试减少batch size或者降低输入图片的分辨率，也可以考虑使用更高显存的显卡或者分布式训练等方法来解决。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值