RuntimeError:无法找到有效的cuDNN算法来运行卷积

Question

我正在用自定义数据集训练yolov5，并在标题中得到错误。我使用的命令:

$ python train.py --img 640 --batch 16 --epochs 3 --data D:customdata.yaml --weights yolov5s.pt 

我有python 3.9, pytorch 1.11和cuda 11.7

Answer 1

请检查您的cuDNN版本。CUDA 11.7似乎是最近的更新，cuDNN可能会遇到问题。尝试使用CUDA 11.6代替cuDNN 8.4.1。