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