【Error】Yolo 训练遇到的错误

最新推荐文章于 2024-07-17 21:03:51 发布
最新推荐文章于 2024-07-17 21:03:51 发布
阅读量1k 收藏
点赞数
分类专栏： Error 文章标签： YOLO
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/WakingStone/article/details/129466995
版权
当遇到RuntimeError提示找不到有效的cuDNN算法运行卷积时，通常是由于GPU内存占用过高。解决方案是调整batch_size，通常建议设为8的倍数，并根据GPU使用情况，如使用Yolov5的s、n、l、x模型时，适当调整以确保训练过程顺利进行。可以使用nvidia-smi命令监控GPU状态并优化batch_size，以减少训练时间。
摘要由CSDN通过智能技术生成

目录

 1.RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

 1.RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

发生报错的原因是gpu内存占用过高

我们需要把bach_size调小，一般建议是8的倍数，内存不够用时尽量调低。

parser.add_argument('--batch-size', type=int, default=16, help='total batch size for all GPUs, -1 for autobatch')

结果运行正常。

使用Yolov5模型时，s和n模型的深度和通道层数都比较小，batch-size设置的稍大一些，gpu可以跑的动，但使用l和x模型跑的时候就不一定可以跑的动了。使用指令nvidia-smi，根据gpu的使用情况调整batch-size的大小，尽量缩短每次训练模型的时间。

悠悠小船长
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
yolo训练自己数据集.txt
05-16
yolo训练自己数据集参考链接集合，亲测好用
手势识别大拇指yolo训练集
06-08
在本“手势识别大拇指YOLO训练集”中，重点聚焦于识别大拇指的手势，这在很多应用场景中具有实际价值，如虚拟现实、智能设备控制以及无障碍交流等。 YOLO（You Only Look Once）是一种实时目标检测系统，它以高效和...
torch.cat 数据类型的坑
qq_43110298的博客
04-13 1822
all_img = torch.tensor([]) # 采用下面的语句读取图片 img = torch.from_numpy(cv2.imread('{}.JPEG'.format(5))).unsqueeze(0) # 此时 img 的维度 （1,224,224,3） # 将img合并入all_img 中 all_img = all_img.cat((all_img,img)) # 报错 RuntimeError: Expected object of scalar type Byte but go.
RuntimeError: expected a non-empty list of Tensors
fghjbjhgb的博客
10-14 7705
这个问题的原因是因为向量中有空，而torch.cat遇到会报错，我加了个try ，跳过了报错。暂时先这样，后边再看看行不行。
RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
最新发布
weixin_44012667的博客
07-17 202
【bug解决】RuntimeError: Unable to find a valid cuDNN algorithm to run convolution-CSDN博客。Pytorch 无法找到有效的cuDNN算法来运行卷积|极客笔记 (deepinout.com)
在运行Yolov8时报错RuntimeError: torch.cat(): expected a non-empty list of Tensors的解决方法
m0_73930473的博客
05-27 1178
单看这些代码还是不太容易发现问题的，只可能把你往torch.cat()这个方向引，但如果眼见会发现在报错之前，有这么两个不起眼的警告（还不是报错，只是警告）到这个时候就要思考文件位置会不会有什么问题（因为我的yolov8是已经跑通了的），回去一看，果不其然，对于文件的命名出错了。正常的命名方法是这样的：images放照片；labels放标签，而我两个文件夹都忘打最后的 ‘s’意思是有0张图片，30000张背景，下面是对没有标签的警告。
YOLOv5本地模型训练报错解决
心之所向
08-04 670
训练过程中，发生下图所示的报错，同时pycharm崩溃。intput后面的路径，就是报错那里，后面给的路径。功能，降级到 Pillow 9.5 解决了该问题。出现下图所示表示执行完毕。这是因为安装了新版本的。
训练YOLOV8时遇到的Runime问题
weixin_43826536的博客
05-08 523
训练yolov8时遇到的runtime问题。
关于yolov10-DDP训练失败的问题及解决办法(24.6.13)
Axh1ne的博客
06-13 939
yolov10-DDP问题解决
快递包裹YOLO训练数据集
04-25
【标题】"快递包裹YOLO训练数据集"指的是一个专门针对快递包裹识别的深度学习模型训练数据集。YOLO（You Only Look Once）是一种实时目标检测系统，它在计算机视觉领域广泛应用，尤其在物体识别方面表现出色。这个...
YOLO2+windows训练配置
12-26
在windows下利用自己的数据，训练YOLO2模型，文档内容一步一步教你如果配置训练数据。
yolo训练 可识别滑块缺口等一系列验证码 易语言源码
04-04
这个压缩包文件中的资源可能包含了一套完整的易语言源码，用于训练YOLO模型以识别特定类型的验证码，例如滑块缺口，以及行人和汽车等对象。 YOLO的核心思想是将图像分类与目标检测结合在一个单一的神经网络中，它能...
Yolov5训练模型时报错RuntimeError: result type Float can t be cast to the desired output type long int
qq_57371368的博客
01-20 1343
这种问题通常发生在 PyTorch 中，因为 PyTorch 张量具有动态数据类型，而在某些情况下，对数据类型的隐式转换可能会导致问题。确保在处理张量时，特别是在构建元组或列表时，要确保数据类型的一致性，以避免此类错误。返回的张量可能包含浮点数，而在你的元组中，它们需要被转换成整数。在 Python 中，将浮点数强制转换为整数可能会导致数据类型错误，因为浮点数不能直接转换为整数。这样就避免了直接修改原始张量，并且在构建元组时，所有的值都是整数类型。函数时，它会原地修改张量，并确保张量中的值在指定的范围内。
使用yolov5训练模型报错：RuntimeError: result type Float can‘t be cast to the desired output type long int
慕溪同学的博客
03-27 449
当我在colab云端训练yolov5模型的时候发现一个错误：RuntimeError: result type Float can’t be cast to the desired output type long int。
YOLOv7 训练报错 IndexError: list index out of range
热门推荐
恐龙先生的博客
07-18 1万+
解决YOLOv7中的报错：IndexError : list index out of range
yolo加载模型出错问题记录
zwl613613613的博客
04-29 569
1、加载时出错 可能时模型加解密时出问题的，release的程序需要用release的加密库 2、preRun出错 保存的模型本身有问题
RuntimeError：Error(s) in loading state dict for YOLOX:
小刘研CV的博客
09-29 2695
RuntimeError：Error(s) in loading state dict for YOLOX:
yolov8使用VFL Loss报错
weixin_47509723的博客
03-16 650
yolov8使用细节
[已解决]Unable to find a valid cUDNN algorithm to run convolution
深度学习炼丹师，偶尔分享炼丹技术!
03-31 1499
问题描述：Unable to find a valid cUDNN algorithm to run convolution可能原因： 1、cuDNN库的配置问题。 2、未指定训练使用的显卡或显卡占用率已满。 3、确定性算法。解决方案：（从下往上试） 1、更新GPU驱动、cuDNN库。 2、禁用cuDNN。 3、更换卷积算法。 4、更换训练用的显卡。 5、禁用确定性算法
用YOLO的时候报RuntimeError: Numpy is not available
07-07
当你在使用YOLO（You Only Look Once）这样的深度学习框架时，可能会遇到`RuntimeError: Numpy is not available`这样的错误，这通常是因为你在运行过程中依赖了NumPy库，但该库并未被正确安装或者在当前环境中找不...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值