【问题标题】:Would cuDNN v6.0 work with TensorFlow currently?cuDNN v6.0 目前可以与 TensorFlow 一起使用吗?
【发布时间】:2017-08-27 12:56:18
【问题描述】:

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

【问题讨论】:

  • 最新版本 (TensorFlow 1.0.1) 不适用于我的 cuDNN v6。(截至 4 月 1 日)。
  • 啊,谢谢。那么 cuDNN 版本可能不向后兼容,并且对于每个新版本,我们需要等待任何基础设施升级/调整以适应新版本......

标签: tensorflow 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

      【讨论】:

        猜你喜欢
        • 1970-01-01
        • 2017-04-29
        • 2016-07-15
        • 1970-01-01
        • 2023-04-08
        • 1970-01-01
        • 1970-01-01
        • 1970-01-01
        • 2011-08-15
        相关资源
        最近更新 更多