最近、いくつかのことをやっていて、Python で Caffe モデルを呼び出すときにエラーが発生することがあります(Caffe のコンパイル時のエラーではありません)。
Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM
Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_BAD_PARAM
インターネットで多くの情報を調べましたが、問題を解決することはできませんでした。
偶然、以前 GitHub のページで見た一つの文を思い出しました。それが私の問題を解決しました。
それは、畳み込み層の convolution_param に engine: CAFFE を追加することです。