【error】RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

问题：

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

解决方法：

减小batch-size的数值即可；

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

 

参考

1. 【问题解决】YOLOv5遇到Unable to find a valid cuDNN algorithm to run convolution；

完