【Caffe】Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM

最近在搞一些事情,有些情况用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

原文:https://blog.csdn.net/renhanchi/article/details/78500665

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容