tensorflow - 2019-03 最新安装的 cudnn 按照 tensorflow apt-get Ubuntu 18.04 说明不再有效。接下来要尝试什么？

Question

在 keras 中使用 Conv1D 获取以下错误：

tensorflow.python.framework.errors_impl.UnknownError：获取卷积算法失败。这可能是因为 cuDNN 初始化失败，因此请尝试查看上面是否打印了警告日志消息。

曾经工作，但我清除并更新到最新的一切。试了几次。

关于什么是坏的任何想法？

编辑：

没有明确说明，但似乎你应该清除所有 nvidia 和 cuda 并确保安装

sudo apt-get install nvidia-410

和 cuda 10-0。

10-1

标签： tensorflowubuntunvidia

Answer 1

我曾经遇到过同样的问题。在 Jupyter 笔记本控制台上，我看到了您提到的错误。在我启动 Jupyter 的终端中，我可以看到以下消息：

2019-03-24 13:27:14.571966：E tensorflow/stream_executor/cuda/cuda_dnn.cc:328] 加载运行时 CuDNN 库：7.0.5 但源编译为：7.4.2。在 CuDNN 7.0 或更高版本的情况下，CuDNN 库主要和次要版本需要匹配或具有更高的次要版本。

这是因为我有 cuDNN 7.0.5 版本，但 Tensorflow 是用 cuDNN 7.4.2 构建的。将 cuDNN 版本升级到 7.5 解决了这个问题。