CuDNN: could not create cudnn hadle: CUDNN_STATUS_NOT_INITIALIZED

最新推荐文章于 2025-06-14 11:36:29 发布

原创最新推荐文章于 2025-06-14 11:36:29 发布 · 3.2k 阅读

1 ·

CC 4.0 BY-SA版权

本文记录了在使用TensorFlow 1.7时遇到的CuDNN初始化失败问题及解决方案。作者最初安装了不匹配的CUDA和CuDNN版本，导致无法正常运行含有CNN的模型。通过调整CuDNN版本至CUDA 9.0对应版本后，问题得到解决。

问题描述：

CuDNN: could not create cudnn hadle: CUDNN_STATUS_NOT_INITIALIZED

Check failed: stream->parent()->GetConvolveAlgoritms( conv_parameters.ShouldIncludeNinogradeNondusedAlgo<T>(), &algoriyms)

已放弃（核心已转储）

环境介绍：

系统：Ubuntu 16.04

CUDA: CUDA-9.0

CuDNN: CuDNN-7.1 for CUDA9.1 (其实写到这里就知道肯定会出问题的)

我在安装tensorflow1.7的时候，使用的CUDA-9.1和CuDNN-7.1 for CUDA9.1，但是实际在import tensorflow的时候，提示的是找不到 libcudart.so.9.0，所以随后将CUDA-9.1卸载，再安装CUDA-9.0，在还没有更改CuDNN的版本下，成功的 import tensorflow。

在跑程序的时候，因为程序中使用到了CNN，就得到了以上的错误，看网上的说法是如果模型没有使用CNN，应该不会报错，但是我没有去尝试了。错误提示的是：CUDNN没有初始化，再下面一句话是“可能驱动版本不足”。我将CUDNN版本换成CUDNN for CUDA-9.0 之后边没有这个错误提示了。所以不是驱动的问题。