基于Anaconda pycharm windows的 tensorflow环境搭建

 
首先按照tensorflow官方教程了解对应版本关系，这里选择最新的

在这个网站下载CUDA Tool kit 11.2，并且在这个网站选择下载cudnn 8.1；注意选择windows x64版本

注意，下载cudnn需要注册账号，并且通过勾选如下图，之后按指示填写问卷成功注册账号即可下载而；而且网站可能上不去，这需要你自己尝试，加速器加速NVIDIA GeForce Experience，VPN等下载好的文件如下（cudnn8.1和cuda11.2.2）

接下来直接照着别人的博客走一遍除了第4步的安装流程，安装完了之后打开anaconda powershell prompt我们自己配置环境，依次输入以下代码即可

conda create -n tensorflow2 python=3.7
conda activate tensorflow2
pip install tensorflow-gpu==2.6.0 -i https://pypi.tuna.tsinghua.edu.cn/simple    # 指定版本为2.6.0，不确定使用清华源是否会影响pip安装结果，在后续使用中确认没有问题，但是一定不能使用conda install tensorflow
conda install opencv matplotlib pillow scipy

注意，打开新的anaconda powershell prompt或另一个cmd窗口之后，使用如下代码验证tensorflow是否安装成功

python
import tensorflow as tf
tf.config.list_physical_devices('GPU')

之后在实际的pycharm运行中，在第一次调用keras时报错，因此根据这个网站指示，降低keras版本为2.6.0

conda list keras    # 确认keras版本确实是2.7.0
pip uninstall keras
pip install keras==2.6.0

至此，环境配置完毕

引用

• [1] 【2021新教程】win10如何安装tensorflow-gpu版本？

• [2] cuda11.2+cudnn8.1+tensorflow2.5.0环境配置

• [3] 关于AlreadyExistsError: Another metric with the same name already exists.的解决方案

• [4] pip安装TensorFlow太慢了怎么办？