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

Question

我们是否特别需要用于 TensorFlow 的 cuDNN v5.1（如建议的那样），或者最新版本 (v6.0) 是否也能正常工作？ cuDNN 版本是否具有向后兼容性？

Answer 1

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

TensorFlow 1.2 may be the last time we build with cuDNN 5.1. Starting with TensorFlow 1.3, we will try to build all our prebuilt binaries with cuDNN 6.0. While we will try to keep our source code compatible with cuDNN 5.1, it will be best effort.

希望下一个 1.3 版本将使用 cuDNN 6.0。尤其是现在，7.0 is right around the corner.

Answer 2

这将是回复的更新答案。

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

pip install tensorflow
pip install --upgrade tensorflow

安装详细信息为 here。

Answer 3

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

P.s。如果您希望它与 cuDNN 5.1 一起使用。您可以安装以前的版本，例如v1.2 在此 link:

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

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

Would cuDNN v6.0 work with TensorFlow currently?

tensorflow

cudnn