Check failed: status == CUDNN_STATUS_SUCCESS (1 vs. 0) CUDNN_STATUS_NOT_INITIALIZED

偏执的太偏执、 2023-01-01 01:44 251阅读 0赞

最近在做onnx模型转换到caffe模型的工作，在onnx=>caffe的时候遇到如下错误：

![20201230141349654.png][]

网上搜索这个问题的解答很多，大多都是使用超级用户运行还有就是重新编译caffe(注释掉cudann)，我实在docker环境下运行的本身就是root用户，重新编译caffe太繁琐，最后在[https://github.com/MTlab/onnx2caffe/issues/25][https_github.com_MTlab_onnx2caffe_issues_25]找到解决方法，亲测可行，在模型prototxt卷积参数部分添加engine: CAFFE即可，也就是修改模型的prototxt中的所有convolution\_param参数添加engine: CAFFE，示例入下：

原始内容：

layer {
      name: "Conv_0"
      type: "Convolution"
      bottom: "input"
      top: "470"
      convolution_param {
        num_output: 64
        bias_term: true
        group: 1
        pad_h: 1
        pad_w: 1
        kernel_h: 3
        kernel_w: 3
        stride_h: 2
        stride_w: 2
        dilation: 1
      }
    }

修改后：

在进行caffemodel转换时采用修改后prototxt的即可。

[20201230141349654.png]: /images/20221120/574e8607b8f040ba82e8cea799cfafea.png
[https_github.com_MTlab_onnx2caffe_issues_25]: https://github.com/MTlab/onnx2caffe/issues/25

Check failed: status == CUDNN_STATUS_SUCCESS (1 vs. 0) CUDNN_STATUS_NOT_INITIALIZED

发表评论取消回复

还没有评论，来说两句吧...

相关阅读

相关 Check failed: status == CUDNN_STATUS_SUCCESS (1 vs. 0) CUDNN_STATUS_NOT_INITIALIZED

相关 Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR

相关 Check failed: mdb_status == 0 (2 vs. 0) No such file or directory

相关 Check failed: status == CUBLAS_STATUS_SUCCESS (13 vs. 0) CUBLAS_STATUS_EXECUTION_FAILED

相关 Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR

相关 win0下caffe训练时出现Check failed: status == CUDNN_STATUS_SUCCESS (1 vs. 0) CUDNN_STATUS_NOT_INITIALIZE的处理

相关 caffe训练出错：Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR

相关 MobileNet-SSD报错Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR

相关 Check failed: error == cudaSuccess (74 vs. 0) misaligned address

相关【解决】Check failed: status == CUDNN_STATUS_SUCCESS (8 vs. 0) CUDNN_STATUS_EXECUTIO