**
小米笔记本安装tensorflow-gpu步骤
**
需要准备相应的软件工具
本人安装的版本如下图:
首先,是下载python3 ,版本按自己的喜好下载。https://www.python.org/ python下载地址,自己选择合适的版本和对应的系统,安装的时候别忘记添加路径到path里面。
如果不知道该下载什么版本的cuda,请打开自己的显卡。
然后点击帮助 -> 系统信息
再点击组件
这里你可以看到自己的显卡支持的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
自己选择安装的地方,上图为默认地址,然后点ok
读取进度条后显示这个界面
然后同意,下一步,打勾即可
安装完后解压cudnn压缩包
将里面的东西覆盖到下面的文件目录下
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
准备工作弄完,可以开始安装tensorflow-gpu了
我安装的方式是 pip install tensorflow-gpu
漫长的等待后提示安装成功
接着便是导入tensorflow看是否报错,没有便是安装成功
自此全部安装完成