如何在 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

正如你提到的关于 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 env 中安装 tf 2.4 并使用 gpu 进行张量操作。

Answer 3

自 2021 年 6 月起，您可以使用 2.4 的标准 conda 安装：

conda install tensorflow-gpu==2.4.1

如@theastronomist 所述，最好使用 conda search tensorflow 查找启用了哪些版本。