YOLOv5遇到Unable to find a valid cuDNN algorithm to run convolution

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即可。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

推薦閱讀更多精彩內(nèi)容