如何在RTX 2070上使用Tensorflow GPU和CUDA 10

Question

几天后，我将用RTX 2070安装我的新电脑。我想使用tensorflow GPU，但我找不到兼容版本的CUDA和tensorflow GPU。据我所知，我需要CUDA 10从RTX的图灵架构的额外计算能力中受益。但关于Tensorflow网站，最新版本的tf(Tensorflow_gpu-1.12.0)仅适用于CUDA 9。我更愿意在windows10上实现这一切，但如果没有其他方法，linux也可以。

我在网上读到两个谣言：1.有一种方法可以编译与CUDA 10一起使用的未发布版本的tf gpu2.他们将在2019年1月发布tf gpu的官方版本(现在几乎结束了)，该版本将支持CUDA 10。

有人能证实其中一个谣言吗(最好有消息来源)，或者告诉我如何才能让一切正常运转？

Answer 1

你需要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:/strong>

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