TensorFlow 2.3 找不到 GPU

【问题标题】：Tensorflow 2.3 doesn't find GPUTensorFlow 2.3 找不到 GPU
【发布时间】：2021-03-21 10:39:49
【问题描述】：

由于某种原因，tensorflow 没有使用我的 GPU。当我运行tf.test.is_built_with_cuda() 时，它返回False。我安装了 tensorflow-gpu，版本 2.3。我还安装了 CUDA 10.1 和 cuDNN 7.65 并具有正确的环境变量。

当我在控制台中运行 nvcc -V 时，它会验证我的 CUDA 是版本 10.1，V10.1.243

当我运行 nvidia-smi 时，它显示我的驱动程序版本是 460.79，而 CUDA 版本是 11.2

我可能需要降级我的 GPU 驱动程序吗？

【问题讨论】：

您的tf.test.is_built_with_cuda() 结果似乎相当确定您没有使用支持 GPU 的 TF 版本。您不需要降级您的 GPU 驱动程序。你几乎从不需要这样做。
我现在尝试卸载 tensorflow 并重新安装 tensorflow-gpu 几次。我该如何解决这个问题？
能分享一下导入tensorflow时显示的日志吗？
@ranka47 如何获取这些日志？

【解决方案1】：

我也有同样的问题，但是当我安装 PyTorch 时，它正确检测到我的 GPU，之前 tf 2.1.0 工作正常。

也许这是一个错误。

我会尝试安装tf 2.1，看看它是否检测到GPU。

2021 年 1 月 7 日更新：尝试通过 conda 安装 tf 2.1.0 版本，但没有成功。

Tensorflow 能够检测到我的 gpu。我从 conda 安装中卸载了 tf，并通过 pip 安装了它。我认为 conda 版本没有检测到 gpu。确保你正在安装的 tensorflow 版本匹配CUDA 和 cudnn 库的要求。

我目前的配置： tensorflow-gpu：2.3.0

CUDA:7.6.5

cudnn:10.1

【讨论】：