gpt4 book ai didi

tensorflow - 支持 Nvidia CUDA Toolkit 9.2

转载 作者:行者123 更新时间:2023-12-04 14:22:48 25 4
gpt4 key购买 nike

Tensorflow-gpu 绑定(bind)到 Nvidia 的 CUDA Toolkit 的特定版本的原因是什么?当前版本似乎专门寻找 9.0,并且不适用于更高版本。例如,我安装了最新的 Toolkit 9.2 并将其添加到路径中,但 Tensorflow-gpu 无法使用它并提示它正在寻找 9.0。

我可以看到不支持主要版本更新而是次要版本?

最佳答案

下载 TF 时,会下载一个预构建的二进制文件。
在构建过程中,TF 硬链接(hard link)到特定版本的 Cuda,因此您不能将其与不同的 cuda 版本一起使用。

如果您想使用新的(或有时较旧的)cuda 版本,您需要从源代码安装 TF (check how here)
或者,如果你真的不想自己构建,请查看这些 repos,还有其他发布特定 TF 二进制文件的,几个例子:

  • https://github.com/mind/wheels
  • https://github.com/yaroslavvb/tensorflow-community-wheels
  • https://github.com/fo40225/tensorflow-windows-wheel

  • 为了您的方便,我在这里添加了每个预构建的 Tensorflow 版本所需的 CUDA + cuDNN 版本:

    (我在这里只写我使用的 TF 版本,也许旧的 TF 版本也使用旧版本的 CUDA)
  • 在 TF v1.5 cuda 8.0 和 cuDNN 6 之前
  • 从:1.5 开始 - 现在针对 CUDA 9 和 cuDNN 7 构建预构建的二进制文件。
  • 关于tensorflow - 支持 Nvidia CUDA Toolkit 9.2,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50605684/

    25 4 0
    Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
    广告合作:1813099741@qq.com 6ren.com