如何在 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。
安装步骤:
- 安装 CUDA 10 和 cuDNN 7.3.1
- 配置Tensorflow并编译
- 使用 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 步:
几天后,我将在我的新电脑上安装 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。
安装步骤:
- 安装 CUDA 10 和 cuDNN 7.3.1
- 配置Tensorflow并编译
- 使用 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 步: