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