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,然后重启，重新测试即可