When we test the yolov5s.pt model for pre-training, the training will report the error Unable to find a valid cuDNN algorithm to run convolution. When you reinstall cudnn, the problem still cannot be solved.
This problem is very simple, it is not that there is a problem with our cudnn. Instead, since the video memory may be insufficient during training, you only need to adjust the batch_size smaller at this time.