几天后,我将为我的新计算机安装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。
有人能证实其中一种谣言(最好是有消息来源),或者告诉我我将如何使其一切正常?
答案 0 :(得分:2)
您是正确的,您需要cuda 10,并且tensorflow-gpu当前不支持它。您需要做的是像第一个谣言一样从源代码编译张量流。
安装步骤:
这里有一些教程来编译张量流。
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: