【Caffe】ステータスのチェックに失敗しました status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM

2017年11月9日#Caffe139

AI 翻訳

この記事はAIを通じて英語から日本語に翻訳されました。原文を表示

AI が生成した要約

最近在使用Python调用Caffe模型时遇到了一个错误，错误信息为"Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM"。在网上查找了很多资料，但是没有解决我的问题。最后在某个Github主页上找到了解决方法，即在卷积层的convolution_param中添加engine: CAFFE。

最近、いくつかのことをやっていて、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 を追加することです。