YOLOv5 Unable to find a valid cuDNN algorithm to run convolution

当使用YOLOv5l.pt进行预训练时,可能会遇到'Unable to find a valid cuDNN algorithm'的错误,即使重装cudnn也无法解决。实际上,问题根源在于模型过大导致显存不足。降低batch_size是有效的解决方法。
摘要由CSDN通过智能技术生成

YOLOv5遇到的bug处理方法
当我们测试yolov5l.pt这个模型进行预训练的时候,会报Unable to find a valid cuDNN algorithm to run convolution这个错误,当你重新安装cudnn后仍然不能解决问题。其实这个问题很简单,并不是我们的cudnn出问题了。而是,由于yolov5l属于大模型。显存可能不足,这时候只需调小batch_size即可。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值