最近在搞一些事情,有些情況用 python 調用 caffe 模型時會報錯(注意不是編譯 caffe 時報的)
檢查失敗: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM
檢查失敗: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_BAD_PARAM
在網上查了很多資料,不過都沒解決我的問題。
意外想到曾經在某個 Github 主頁上看到過一句話。最終解決了我的問題。
就是在卷積層 convolution_param 中添加 engine: CAFFE