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

Question

我正在尝试使用官方网站中提到的pip install tensorflow安装最新版本的 TensorFlow 2 。 我有最新版本的Ubuntu 20.04 LTS 。 我的笔记本电脑有Nvidia GeForce GTX 1050ti显卡。 但是当我安装 Nvidia 驱动程序时，它只安装最新版本 - 455.45 及其CUDA 版本 - 11.1。

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

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

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

Answer 1

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

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

conda create --name tf_gpu tensorflow-gpu

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