如何在anaconda上安装tensorflow 2.4>？

Question

conda的安装只针对TF2.2.0，因为在anaconda上似乎没有cudnn 8.0

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

有人知道获得tf 2.4的方法吗>；=在Python环境中运行

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，然后使用TF2.3。这将支持基于gpu的张量运算。在这种情况下，您还必须降级其他环境中的tf（例如Pycharm IDE中的conda env），因为CUDA 10。将不支持降级的tf版本

或者你可以等到水蟒支持TF2.4。在这种情况下，你不需要改变任何地方。请稍等，在conda环境中安装TF2.4，并将gpu与tensor操作一起使用

Answer 3

从2021年6月起，您可以在2.4中使用标准conda安装：

conda install tensorflow-gpu==2.4.1

正如@theastronomist提到的，最好使用conda search tensorflow来查找启用了哪些版本