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 所需的所有正確軟件。