cuDNN v6.0目前是否适用于TensorFlow?

时间:2017-04-02 17:37:02

标签: tensorflow cudnn

我们是否特别需要TensorFlow的cuDNN v5.1(如建议的那样),或者最新版本(v6.0)是否也能正常工作? cuDNN版本中是否存在向后兼容性?

3 个答案:

答案 0 :(得分:2)

不,最新的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

答案 1 :(得分:2)

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

P.S。如果您希望它与cuDNN 5.1一起使用。你可以安装以前的版本,例如v1.2就在这个链接上:

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

答案 2 :(得分:1)

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

我已将tensorflow版本升级为1.2.1,然后cudnn 6.0似乎没有问题。我使用了pip

pip install tensorflow
pip install --upgrade tensorflow 

安装详情为here