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

How to use Tensorflow GPU with CUDA 10 on RTX 2070?

几天后,我将在我的新电脑上安装 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月发布正式版tf-gpu(现在快结束了),支持CUDA 10.

有人可以证实其中一个谣言(最好有来源)或者告诉我如何才能让它全部发挥作用吗?

你说得对,你需要 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 wheels,从而跳过第 2 步:

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