如何在 anaconda 上安裝 tensorflow 2.4>？

Question

conda 安裝僅適用於 tf 2.2.0，因為 anaconda 上似乎沒有 cudnn 8.0。

https://anaconda.org/anaconda/tensorflow-gpu

有誰知道在 anaconda 環境中運行 tf 2.4>= 的方法？

Answer 1

目前 conda 不支持安裝 tf>2.3.0

$ conda search tensorflow

...
tensorflow                     2.2.0 mkl_py37h6e9ce2d_0  pkgs/main           
tensorflow                     2.2.0 mkl_py38h6d3daf0_0  pkgs/main           
tensorflow                     2.3.0 eigen_py37h189e6a2_0  pkgs/main           
tensorflow                     2.3.0 eigen_py38h71ff20e_0  pkgs/main           
tensorflow                     2.3.0 mkl_py37h0481017_0  pkgs/main           
tensorflow                     2.3.0 mkl_py38hd53216f_0  pkgs/main

順便說一句conda-gpu在 conda 上最多僅支持2.2.0

正如前面提到的答案，您可以在 conda 環境中使用pip install tensorflow 。

Answer 2

截至 2021 年 6 月，您可以使用 2.4 的標准 conda 安裝：

conda install tensorflow-gpu==2.4.1

正如@theastronomist 提到的，最好使用conda search tensorflow來查找啟用的版本。

Answer 3

正如您提到的有關 CUDA 的問題，我假設您想升級 tensorflow 以匹配 CUDA 版本和 tf 版本。 我認為您的 cuda 版本是 11.x。

解決方法有兩種，一是可以卸載CUDA驅動，安裝10.x再使用tf 2.3。 這將支持基於 gpu 的張量操作。 在這種情況下，您還必須在其他環境中降級 tf（例如 Pycharm IDE 中的 conda env），因為 CUDA 10. 將不支持降級的 tf 版本。

或者你可以等到 anaconda 支持 tf 2.4。 在這種情況下，您無需在任何地方更改任何內容。 等一下，在你的 conda 環境中安裝 tf 2.4 並使用 gpu 和張量操作。