YOLOv5遇到的bug處理方法
當我們測試yolov5l.pt這個模型進行預訓練的時候扇商,會報Unable to find a valid cuDNN algorithm to run convolution這個錯誤蛛蒙,當你重新安裝cudnn后仍然不能解決問題呜笑。其實這個問題很簡單亏镰,并不是我們的cudnn出問題了。而是免姿,由于yolov5l屬于大模型排惨。顯存可能不足,這時候只需調(diào)小batch_size即可幌氮。
當我們測試yolov5l.pt這個模型進行預訓練的時候扇商,會報Unable to find a valid cuDNN algorithm to run convolution這個錯誤蛛蒙,當你重新安裝cudnn后仍然不能解決問題呜笑。其實這個問題很簡單亏镰,并不是我們的cudnn出問題了。而是免姿,由于yolov5l屬于大模型排惨。顯存可能不足,這時候只需調(diào)小batch_size即可幌氮。