如何在 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 和张量操作。