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

Question

电脑系统是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)请问该如何解决

Answer 1

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

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