m0_74337277 2023-02-10 00:52 采纳率: 40%
浏览 73
已结题

使用yolov4训练时出现报错: cuDNN没有找到卷积的FWD算法

电脑系统是win11,CUDA版本是11.6,cudnn版本是8.6.0,环境配置完成时,使用yolo官网的权重文件可以正常对图片进行标注,后面训练自己的数据集时,我在labelImg对图片标注有4个标签分类,yolov4.cfg voc.data voc.names这三个文件都进行了相应的配置(附图1,2,3,4),开始训练时,出现了报错:cuDNN没有找到卷积的FWD算法(训练时使用的命令以及出现的错误,如图5,6)请问该如何解决

img

img

img

img

img

img

  • 写回答

6条回答 默认 最新

  • 爱晚乏客游 2023-02-10 10:48
    关注

    img


    你这显示你的DarkNet中的cudnn是8.1.1的版本,这个和你说的8.6对不上。不过按理来说cudnn的影响不会很大(只要不是cudnn7.6.5和cudnn8.x这种大版本大区别就基本上没啥大影响)

    img


    从你这张报错可以看到,模型没有加载完成,再看你的显卡是桌面的3050 ti,考虑下4G显存不足的问题导致模型无法正确加载进显卡,也会报这个错误。
    按照我之前的训练来看,608x608大小的图片yolov4,差不多要7G多的显存(3070接近满载的状态,扣掉一些系统占用,也要7G以上的显存),所以我估计你这个最大的可能就是显存不足,或者你把batch和sub都设置为1,图片大小设置为128看下能否加载,不行的话只能试试其他小一点的模型(yolov4-tiny)能否完成整个加载过程。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(5条)

报告相同问题?

问题事件

  • 系统已结题 2月18日
  • 已采纳回答 2月10日
  • 赞助了问题酬金15元 2月10日
  • 创建了问题 2月10日

悬赏问题

  • ¥60 版本过低apk如何修改可以兼容新的安卓系统
  • ¥25 由IPR导致的DRIVER_POWER_STATE_FAILURE蓝屏
  • ¥50 有数据,怎么建立模型求影响全要素生产率的因素
  • ¥50 有数据,怎么用matlab求全要素生产率
  • ¥15 TI的insta-spin例程
  • ¥15 完成下列问题完成下列问题
  • ¥15 C#算法问题, 不知道怎么处理这个数据的转换
  • ¥15 YoloV5 第三方库的版本对照问题
  • ¥15 请完成下列相关问题!
  • ¥15 drone 推送镜像时候 purge: true 推送完毕后没有删除对应的镜像,手动拷贝到服务器执行结果正确在样才能让指令自动执行成功删除对应镜像,如何解决?