Tensorflow-gpu 不适用于 Nvidia 驱动程序 455.45 和 CUDA 版本 - UBUNTU 20.04 上的 11.1

Question

我正在尝试使用 pip install tensorflow 安装最新版本的 TensorFlow 2，正如 official website 中所述。我有最新版本的 Ubuntu 20.04 LTS。我的笔记本电脑有 Nvidia GeForce GTX 1050ti 显卡。但是当我安装 Nvidia 驱动程序时，它只安装了最新的 version - 455.45 及其 CUDA version - 11.1.

然后我安装了 cuda-toolkit 10.1 和 CuDNN7.6。 但 TensorFlow 仍然无法在我的系统上检测到 GPU。

帮我在最新版本的 Ubuntu 和 Nvidia Driver 上安装 TensorFlow。

如果您发现任何显示正确安装方法的博文，请分享。谢谢。

Answer 1

只需简单地进行 CUDA 的全新安装（首先删除其所有痕迹，然后尝试重新安装正确的版本，即10.1）。

或者：- 您可以在您的计算机上安装 Miniconda 或 Anaconda，然后在 cmd 上运行此命令：-

conda create --name tf_gpu tensorflow-gpu

它将自动下载并安装在 GPU 上运行 TensorFlow 所需的所有正确软件。