您的位置:首页 > Web前端

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

2017-11-10 16:00 603 查看
最近在搞一些事情,有些情况用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
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐