【tensorflow安装gpu版本】记录下

由于每次安装都要去翻看两个博客，太麻烦，这里做个记录总结。

参考的博客：

参考1
参考2

步骤

  1. 先去tensorflow官网找到对应的tensorflow-python-cuDNN-cuda版本
    在这里插入图片描述
    （这里选择了第三行，tensorflow2.4.0+python3.6-3.8+cuDNN8.0+cuda11.0）

  2. 选择好版本后，在创建的环境中使用conda命令安装

#安装cuda，先用conda search cuda搜索版本
conda install cudatoolkit=11.0.221
#安装cuDNN 同样使用conda search搜索
conda install cudnn==8.0.5.39 -c conda-forge
#安装tensorflow-gpu
pip install tensorflow-gpu==2.4.0

如果使用conda search直接搜索不到所需要版本，使用conda search cudnn -c conda-forge进行搜索，安装时的命令同样带上-c conda-forge，如果cuda找不到版本也一样操作
3. 安装完成
在这里插入图片描述