如何在 RTX 2070 上使用带有 CUDA 10 的 Tensorflow GPU？

Question

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

我在互联网的某个地方读到了两个谣言： 1. 有一些方法可以编译与 CUDA 10 一起使用的未发布版本的 tf-gpu 2.他们将在2019年1月（现在差不多结束）发布支持CUDA 10的正式版tf-gpu。

谁能证实这些谣言之一（最好有来源）或告诉我如何才能让这一切正常运行？

Answer 1

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

安装步骤：

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

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

Windows：

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