tensorflow2.1.0 安装及GPU配置（Windows/Python）

本文链接：https://blog.csdn.net/qq_40719787/article/details/105189464

tensorflow官网：https://tensorflow.google.cn/

tensorflow2.1.0 Python pip 安装

pip install --upgrade pip
pip install tensorflow

这两条命令执行结束之后，其实tensorflow2.1.0已经安装完成了，不过在Pycharm中会提示未配置GPU的问题。

tensorflow GPU配置

在这里插入图片描述
这是官方给出的一套准备流程，其中CUDA® 工具包和cuDNN SDK需要我们额外安装，第三项会默认安装的，如果你的显卡驱动版本过低也需要自己进行更新。

CUDA® 工具包的获取：【官网下载地址】

这里要注意版本的问题，现在tensorflow官网写的使10.1版本
在这里插入图片描述
点击之后选择你的电脑配置项，我直接把整个文件都下载下来了，这样安装起来也比较方便。

在这里插入图片描述
安装过程我是按照它的默认设置都安装在了C盘，如果C盘空间不多了，选择了自定义安装切记记住最终的安装位置，因为后面需要配置环境变量。

cuDNN SDK下载：【点这里跳过去】

在这里插入图片描述
这里要求具有NVIDIA开发人员计划的成员资格，如果没有注册一个就行。

下载完成之后直接解压就可以。
解压成功之后会有一个cuda文件夹
将这个文件夹移动到C:\tools目录下（没有的话就创建下tools文件夹）

环境变量配置

官网配置方式：

    SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;%PATH%
    SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\libx64;%PATH%
    SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include;%PATH%
    SET PATH=C:\tools\cuda\bin;%PATH%

我的配置方式使按照Java环境那样配置的

我的电脑–(右键)–》属性–》高级系统设置-》环境变量（N）
找到Path 将下面四条添加进去就行，如果自定义的安装位置，注意一下路径，以后可能还得注意一下版本号！

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include
C:\tools\cuda\bin