hans

hans

【Caffe】检查失败 状态 == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM


最近在搞一些事情,有些情况用 python 调用 caffe 模型时会报错(注意不是编译 caffe 时报的)

检查失败:status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM

检查失败:status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_BAD_PARAM

在网上查了很多资料,不过都没解决我的问题。

意外想到曾经在某个 Github 主页上看到过一句话。最终解决了我的问题。

就是在卷积层 convolution_param 中添加 engine: CAFFE

加载中...
此文章数据所有权由区块链加密技术和智能合约保障仅归创作者所有。