win10系统下GTX1080TI显卡配置tensorflow-gpu运行环境(都是泪系列)
原来在Ubuntu16.04系统下装过一次运行环境,现在因为项目需求在win10下重装还是经过了一番波折,在这记录下正确的安装方式,分享出来!!
安装准备
首先你得知道安装tensorflow-gpu需要安装哪些环境或者工具,给出一张版本适配图有助于理解。
从图上可以看出,我们需要安装做4项工作即安装Python、安装c编译器及其环境、安装cudnn和cuda。下图是我安装的工具版本。我已经上传到百度云,有需要可以直接下载链接:https://pan.baidu.com/s/1u9DMthqBRIMMhFgdWru1xA 提取码:gsz6 复制这段内容后打开百度网盘手机App,操作更方便哦
。
安装步骤详解
安装MSVC 2015update 3和编译工具
这个步骤只需要安装好vc2015即可,安装目录没有要求。百度云链接中的VC2015软件需要在线下载安装,过程比较慢只需要等就好了。
安装cuda 9.0
这个cuda软件一共有5个可执行文件,先安装cuda_9.0.176_win10.exe主程序,后面四个都是补丁,按照顺序一个一个安装即可。默认安装目录,其他的没尝试过。
安装后还有一个问题就是配置环境变量,即将cuda9.0的安装目录配置到系统环境变量中,使得其他应用软件可以直接查询到该目录。具体做法是首先找到安装目录,如果是默认目录则应该是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 ,将该目录复制,然后右击此计算机打开属性后出现此窗口。
然后打开高级系统设置找到环境变量按钮,如下图
点击环境变量按钮,后找到Path变量,然后双击
然后点新建将刚才的目录复制上去。
测试下cuda是否安装成功,打开CMD,输入nvcc -V
,如果有反馈则说明安装成功。
然后就是安装cudnn 我下载的cudnn v7.1,该过程只需要将cudnn文件按照存放的目录复制到指定的cuda目录中。
Anaconda安装过程
首先我安装的Anaconda3-4.2.0(python3.5),安装过程没什么好说的一路点下一步,其中有个窗口让配置到系统变量,打钩即可。
tensorflow安装过程
打开Anconda prompt,系统的原因最好以管理员身份运行。最近国内源许多都失效了,但还是有能用的。pip install 包名-i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
我安装到是tensorflow-gpu 1.9.0,因此直接输入pip install tensorflow-gpu==1.9.0 -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
等待安装即可。
安装好之后可以简单测试下是否安装成功。