时间戳

首页 > 解决方案 > cuDNN 初始化失败

python - cuDNN 初始化失败

问题描述

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 索泰 gtx 1070 ti

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

标签： pythontensorflow

解决方案

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

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

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

完成后再次尝试运行代码

推荐阅读