tensorflow2.1.0 安装及GPU配置(Windows/Python)

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

tensorflow2.1.0 Python pip 安装

pip install --upgrade pip
pip install tensorflow

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

tensorflow GPU配置

tensorflow2.1.0 安装及GPU配置(Windows/Python)_第1张图片
这是官方给出的一套准备流程,其中CUDA® 工具包cuDNN SDK需要我们额外安装,第三项会默认安装的,如果你的显卡驱动版本过低也需要自己进行更新。

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

这里要注意版本的问题,现在tensorflow官网写的使10.1版本
tensorflow2.1.0 安装及GPU配置(Windows/Python)_第2张图片
点击之后选择你的电脑配置项,我直接把整个文件都下载下来了,这样安装起来也比较方便。

tensorflow2.1.0 安装及GPU配置(Windows/Python)_第3张图片
安装过程我是按照它的默认设置都安装在了C盘,如果C盘空间不多了,选择了自定义安装切记记住最终的安装位置,因为后面需要配置环境变量。

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

tensorflow2.1.0 安装及GPU配置(Windows/Python)_第4张图片
这里要求具有NVIDIA开发人员计划的成员资格,如果没有注册一个就行。
tensorflow2.1.0 安装及GPU配置(Windows/Python)_第5张图片
下载完成之后直接解压就可以。
解压成功之后会有一个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

验证GPU配置
tensorflow2.1.0 安装及GPU配置(Windows/Python)_第6张图片

这里如果全都是Successfully表示就可以了,如果没有的话尝试重启Pycharm,或者检查环境变量是否配置正确。

配置成功之后你再运行你的项目你就会发现GPU它动了…QAQ…

你可能感兴趣的:(python,tensorflow,python)