Tensorflow 1.15无法用Cuda10.1检测到gpu

Question

我已经安装了tensorflow 2.2.0和tensorflow 1.15.0 （通过pip install tensorflow-gpu==1.15.0 ）。 tensorflow 2 安装在Anaconda 3的基础环境中，而 tensorflow 1 安装在单独的环境中。

tensorflow 2.2.0可以通过简单测试识别gpu：

if tf.test.gpu_device_name():
    print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))

//output: Default GPU Device: /device:GPU:0

但是tensorflow 1.15.0无法检测到gpu。

供您参考，我的系统环境是 python + cuda 10.1 + vs 2015。

Answer 1

tensosflow版本1.15.0到1.15.3（最新版）都是针对Cuda 10.0编译的。 将 cuda 10.1 降级到 cuda 10.0 解决了这个问题。

还要注意 python 版本。 建议为特定的 python 版本安装 tensorflow .whl文件（如https://nero-mirror.stanford.edu/pypi/simple/tensorflow-gpu/ 所列）。 至于安装，参见How do I install a Python package with a.whl file?

Answer 2

Tensorflow 1.15 期望 cuda 10.0，但我设法通过使用 Anaconda 安装以下软件包使其与 cuda 10.1 一起工作： cudatoolkit (10.0) 和cudnn (7.6.5)。 所以，运行后

conda install cudatoolkit=10.0
conda install cudnn=7.6.5

tensorflow 1.15 能够找到并使用 GPU（正在使用 cuda 10.1）。

PS：我知道你的环境是基于 Windows 的，但是这个问题在谷歌上弹出，因为同样的问题发生在 Linux（我测试了这个解决方案）。 在 Windows 上也可能有用。

Answer 3

您是否尝试安装 Anaconda？ 它会下载所有要求，只需点击几下即可轻松完成。

Answer 4

可能与TF、Cuda、CuDNN的版本兼容有关。 这篇文章对此进行了彻底的讨论。