【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


猜你喜欢

转载自blog.csdn.net/renhanchi/article/details/78500665
今日推荐