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

Posted 2023-04-15

【中文标题】cuDNN v6.0 目前可以与 TensorFlow 一起使用吗？

【英文标题】：Would cuDNN v6.0 work with TensorFlow currently?

【发布时间】：2017-08-27 12:56:18

【问题描述】：

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

【问题讨论】：

最新版本 (TensorFlow 1.0.1) 不适用于我的 cuDNN v6。（截至 4 月 1 日）。

啊，谢谢。那么 cuDNN 版本可能不向后兼容，并且对于每个新版本，我们需要等待任何基础设施升级/调整以适应新版本......

【参考方案1】：

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

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

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

【参考方案2】：

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

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

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

【参考方案3】：

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

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

pip install tensorflow
pip install --upgrade tensorflow 

安装详情为here。