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