小米笔记本安装tensorflow-gpu步骤

**

小米笔记本安装tensorflow-gpu步骤

**
需要准备相应的软件工具
本人安装的版本如下图:
小米笔记本安装tensorflow-gpu步骤_第1张图片

首先,是下载python3 ,版本按自己的喜好下载。https://www.python.org/ python下载地址,自己选择合适的版本和对应的系统,安装的时候别忘记添加路径到path里面。

如果不知道该下载什么版本的cuda,请打开自己的显卡。
小米笔记本安装tensorflow-gpu步骤_第2张图片
然后点击帮助 -> 系统信息
小米笔记本安装tensorflow-gpu步骤_第3张图片
再点击组件
小米笔记本安装tensorflow-gpu步骤_第4张图片
这里你可以看到自己的显卡支持的cuda版本,按着这个大版本下载一般就没问题。
接着下载CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive

CuDNN下载地址:https://developer.nvidia.com/rdp/cudnn-download(下载CuDNN需要注册账号,注册过程比较简单)

如过自己电脑没有安装Microsoft Build Tools 2015的话就需要安装了,因为tensorflow-gpu需要某些他上面的dll组件
Microsoft Build Tools 2015下载地址:https://www.microsoft.com/en-us/download/details.aspx?id=48159

在这里插入图片描述
vc_redist.x64下载地址:https://www.microsoft.com/en-us/download/details.aspx?id=48145

全部下载

开始安装

python安装自己百度
下面两个直接点下一步就可以安装好:
vc_redist.x64
Microsoft Build Tools 2015
在这里插入图片描述
开始安装cuda
小米笔记本安装tensorflow-gpu步骤_第5张图片
自己选择安装的地方,上图为默认地址,然后点ok
读取进度条后显示这个界面
小米笔记本安装tensorflow-gpu步骤_第6张图片
然后同意,下一步,打勾即可
安装完后解压cudnn压缩包
将里面的东西覆盖到下面的文件目录下
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

准备工作弄完,可以开始安装tensorflow-gpu了
我安装的方式是 pip install tensorflow-gpu

漫长的等待后提示安装成功
接着便是导入tensorflow看是否报错,没有便是安装成功
在这里插入图片描述
自此全部安装完成

你可能感兴趣的:(学习)