cuDNN 初始化失败

Question

tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
  (0) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
     [[{{node conv2d_1/convolution}}]]
     [[norm_dense_1/Softmax/_3605]]
  (1) Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
     [[{{node conv2d_1/convolution}}]]

尝试从 Github 运行项目时遇到此错误。我已尝试在新的 conda 环境中从头开始重新安装所有程序。 似乎无法弄清楚是什么问题。

GPU Zotac gtx 1070 ti

我目前的软件安装如下：-

• Python 3.6
• Tensorflow GPU 1.15
• CUDA 10
• cuDNN 7.4

Answer 1

确保您拥有 CUDA Toolkit 7.5 请打开一个新的 session 然后尝试加载 tensorflow

你能在没有任何错误的情况下加载 tensorflow 吗？ 确保正在加载所有 ddl cudard64_101.ddl 和其他文件。 如果 tensorflow 没有问题，则运行以下命令

physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)

完成后再次尝试运行代码

Answer 2

我有同样的 output。 我的问题是我不小心为 cuda 设置了两个不同的环境变量。 PATH 有 2 个位置用于 cuda 文件夹，这对我来说是个问题。 删除其中一个解决了我的问题。