tensorflow - TensorFlow 2.3 找不到 GPU

Question

出于某种原因，张量流没有使用我的 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 驱动程序吗？

标签： tensorflowinstallationgpu

Answer 1

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

也许这是一个错误。

我将尝试安装tf 2.1并查看它是否检测到 GPU。

更新 1/7/2021：尝试通过 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