0

几天后，我将使用 RTX 2070 设置我的新计算机。我想使用 tensorflow GPU，但我找不到 CUDA 和 Tensorflow GPU 的兼容版本。据我所知，我需要 CUDA 10 才能受益于 RTX 图灵架构的额外计算能力。但是关于 Tensorflow 网站，最新版本的 tf (tensorflow_gpu-1.12.0) 仅适用于 CUDA 9。我希望它能够在 Windows 10 上运行，但如果没有其他方法，linux 也可以。

在互联网的某个地方，我读到了两个谣言：1. 有一些方法可以编译一个未发布的 tf-gpu 版本，它与 CUDA 10 一起工作 2. 他们将在 2019 年 1 月发布官方版本的 tf-gpu（几乎结束了现在），它将支持 CUDA 10。

有人可以证实其中一个谣言（最好有消息来源）或告诉我如何才能让这一切正常工作吗？

4

1 回答 1

3

您是正确的，您需要 cuda 10 并且 tensorflow-gpu 目前不支持它。您需要做的是从源代码编译 tensorflow，就像您的第一个谣言一样。

安装步骤：

  1. 安装 CUDA 10 和 cuDNN 7.3.1
  2. 配置 Tensorflow 并编译
  3. 使用 pip 安装 .whl 包

这里有一些编译 tensorflow 的教程。

视窗：

https://www.pytorials.com/how-to-install-tensorflow-gpu-with-cuda-10-0-for-python-on-windows/2/

Ubuntu：

https://medium.com/@saitejadommeti/building-tensorflow-gpu-from-source-for-rtx-2080-96fed102fcca

https://towardsdatascience.com/how-to-make-tensorflow-work-on-rtx-20xx-series-73eb409bd3c0

或者 ，您可以在此处找到预先构建的 tensorflow 轮子，从而跳过第 2 步：

https://github.com/fo40225/tensorflow-windows-wheel

于 2019-01-28T20:39:27.367 回答