如何使用Python在Windows上正確安裝Tensorflow？

Question

我正在嘗試在 python-conda 環境中將 tensorflow 與我的 PC GPU（Nvidia RTX 3070Ti）一起使用。 我正在解決來自 kaggle 的一個小圖像分類問題。 我已經在 google-collab 中解決了它，但現在我有興趣在我的本地機器上解決它。 但是 TF 在本地無法正常工作，我不知道為什么。 我已經閱讀了大量的解決方案，但還沒有幫助。

我遵循本指南並始終安裝正確版本的 TF 和 CUDA： https://www.tensorflow.org/install/source_windows

cuda-toolkit 10.1, cudnn 7.6, tf-gpu 2.3, python 3.8

我還為視頻卡安裝了最新的 NVidia 驅動程序。

我試過的：

1. 我已經從 nvidia 站點安裝了正確版本的 CUDA-toolkit 和 CUDnn。 我已正確安裝它並將所需的一切包含到 PATH 中。 我已經檢查過 - MS Visiual Studio 找到 CUDA 和 CUDnn 並且可以使用它。 我已經使用 conda 在我的環境中安裝了正確版本的 Tensorflow-GPU。

結果： TF 找不到我的 GPU 並且只使用 CPU。

2. 我已經刪除了所有 CUDA 和 CUDAnn 驅動程序。 我已經在我的 conda 環境中安裝了 CUDA-toolkit、CUDnn 和 Tensorflow-GPU python 包。

結果： TF 識別出我的 GPU 並使用它： 但是在 DNN 訓練期間發生錯誤： Failed to launch ptxas Relying on driver to perform ptx compilation. Modify $PATH to customize ptxas location. Failed to launch ptxas Relying on driver to perform ptx compilation. Modify $PATH to customize ptxas location. 訓練非常糟糕——准確性非常低而且沒有提高。

當我在 google-collab 上使用完全相同的代碼和數據時，一切都進行得很順利——我在第 5 個 epoch 上獲得了大約 90% 的准確率。

3. 我試過 tf 2.1 和相關的 cuda 和 cudnn，但結果仍然相同！

4. 我嘗試安裝 cudatoolkit-dev，但它並沒有幫助解決 ptxas 問題。

我即將放棄並使用 PyTorch 而不是 Tensorflow。

Answer 1

所以這對我有用：

1. 創建 3.9 python 環境
2. 從“Esri”安裝 cuda 和 tensorflow 包：

 conda install -c esri cudatoolkit conda install -c esri cudnn conda install -c esri tensorflow-gpu

3. 然后安裝tensorflow-hub：

 conda install -c conda-forge tensorflow-hub

它會從之前的步驟中降級安裝，但它可以工作。 也許首先安裝 tensorflow-hub 可以幫助避免它，但我沒有測試它。