Windows10搭建Tensorflow2 gpu环境(2021.11.15)
1.Anaconda创建tensorflow gpu环境
conda create -n tf2-gpu1 python=3.7
2.查看NVIDIA驱动版本
右键->NVIDIA控制器
我电脑最新刚买的，可以看到对应的NVIDIA驱动版本是11.4。
3.准备各版本
CUDA Toolkit 11.4.0、cuDNN v8.2.x
tensorflow-gpu:2.7.0(最新版)、Anaconda:最新版
注意：下载CUDA Toolkit 11.4.0即可，不能下载高于此版本的。也不能按照网上教程下载CUDA10.x版本的。否则会出现类似如下问题：cudart64_101.dll not found（后面会详细说）
4.安装CUDA和cuDNN
(1)安装CUDA
CUDA下载地址： https://developer.nvidia.com/cuda-toolkit-archive
安装CUDA过程：
Visual Studio Integration不要选，怕和本地环境冲突。
这个也不要选，本地已经有这个，就不用安装了。
接下来等安装完成即可。安装完毕后，按Win+R打开Windows并输入cmd，随后输入nvcc -V
(2)下载cuDNN神经网络加速库
cuDNN下载地址：https://developer.nvidia.com/rdp/cudnn-archive
下载完毕后将cuda文件复制到CUDA的安装目录下，如图所示
(3)配置环境变量
将cuDNN神经网络加速库添加到环境变量中，如图所示
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\cuda\bin
其他几个没有，也加上，一般CUDA安装好了会默认加上，不用手动加
5.安装Tensorflow
在tenforflow-gpu环境下输入：pip install -U tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
测试tensorflow-gpu是否安装成功：
import tensorflow as tf
tf.__version__
tf.test.is_gpu_available()
如图所示，即证明tensorflow-gpu安装成功。
在pycharm配置环境
问题记录：
问题一：找不到cudart的dll文件
在cmd中就出现如下问题：cudart64_101.dll not found。说明版本不对，仔细检查版本。我之前安装了CUDA10.2，出现了这个问题。然后我删除了这个版本，查看本地NVIDIA驱动版本是11.4，于是下载对应的11版本。
问题二：在cmd测试正常，但是在pycharm中出现了：cudart64_101.dll not found。这是环境变量的问题，将系统配好的环境变量加入到IDE的环境变量中，让IDE能找到CUDA的安装目录即可。
在环境变量没问题的情况下，我的解决方案：关掉pycharm,然后重启，重新测试即可