【Caffe】Check failed: status == C

2019-05-24  本文已影响0人  追忆_0e47

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

上一篇下一篇

猜你喜欢

热点阅读