YOLOv5遇到的bug處理方法
當(dāng)我們測試yolov5l.pt這個模型進(jìn)行預(yù)訓(xùn)練的時候,會報Unable to find a valid cuDNN algorithm to run convolution這個錯誤,當(dāng)你重新安裝cudnn后仍然不能解決問題。其實(shí)這個問題很簡單,并不是我們的cudnn出問題了。而是,由于yolov5l屬于大模型。顯存可能不足,這時候只需調(diào)小batch_size即可。
當(dāng)我們測試yolov5l.pt這個模型進(jìn)行預(yù)訓(xùn)練的時候,會報Unable to find a valid cuDNN algorithm to run convolution這個錯誤,當(dāng)你重新安裝cudnn后仍然不能解決問題。其實(shí)這個問題很簡單,并不是我們的cudnn出問題了。而是,由于yolov5l屬于大模型。顯存可能不足,這時候只需調(diào)小batch_size即可。