如何在RTX 2070上将Tensorflow GPU与CUDA 10结合使用?

时间:2019-01-23 20:06:41

标签: python tensorflow installation rtx

几天后,我将为我的新计算机安装RTX 2070。 我想使用tensorflow GPU,但找不到兼容的CUDA和Tensorflow GPU版本。 据我所知,我需要CUDA 10才能从RTX的Turing架构的额外计算能力中受益。但是关于Tensorflow网站,最新版本的tf(tensorflow_gpu-1.12.0)仅适用于CUDA 9。 我希望所有这些都可以在Windows 10上正常运行,但是如果没有其他方法,Linux也可以正常工作。

在互联网上的某个地方,我读到了两个谣言: 1.有一些方法可以编译与CUDA 10一起使用的未发布的tf-gpu版本 2.他们将在2019年1月发布tf-gpu的正式版本(将近结束),该版本将支持CUDA 10。

有人能证实其中一种谣言(最好是有消息来源),或者告诉我我将如何使其一切正常?

1 个答案:

答案 0 :(得分:2)

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

安装步骤:

  1. 安装CUDA 10和cuDNN 7.3.1
  2. 配置Tensorflow并对其进行编译
  3. 使用pip安装.whl软件包

这里有一些教程来编译张量流。

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