tensorflow - cuDNN v6.0 目前可以与 TensorFlow 一起使用吗？

Question

我们是否特别需要用于 TensorFlow 的 cuDNN v5.1（如建议的那样），还是最新版本（v6.0）也可以工作？cuDNN 版本是否存在向后兼容性？

Answer 1

不，最新的 1.2 版本不支持 cuDNN 6.0。但有一个希望：官方发布说明告诉以下内容：

TensorFlow 1.2 可能是我们最后一次使用 cuDNN 5.1 构建。从 TensorFlow 1.3 开始，我们将尝试使用 cuDNN 6.0 构建所有预构建的二进制文件。虽然我们会尽量保持我们的源代码与 cuDNN 5.1 兼容，但这将是最大的努力。

所以希望下一个 1.3 版本将使用 cuDNN 6.0。尤其是现在，7.0 指日可待。

Answer 2

今天（2017 年 8 月 21 日）我安装了最新的 Tensorflow 版本 v1.3，我可以确认它需要cuDNN v6.0 并且不适用于 v5.1。它实际上会要求库 libcudnn.so.6 而不是 libcudnn.so.5

Ps 如果您希望它与 cuDNN 5.1 一起使用。您可以在此链接上安装以前的版本，例如 v1.2：

https://www.tensorflow.org/versions/r0.12/get_started/os_setup#virtualenv_installation

Answer 3

这将是Salvador Dali回应的更新答案。

我已经将tensorflow版本升级到1.2.1，然后cudnn 6.0似乎没有问题。我用了pip

pip install tensorflow
pip install --upgrade tensorflow 

安装细节在这里。