笔者亲试很多次,在自己笔记本(联想拯救者r720,1050Ti显卡)以及深度学习工作站(1080Ti显卡)上安装TensorFlow-GPU都是一次成功,所以在这里分享一下全过程,以帮助大家成功安装TensorFlow,避免采坑(我是踩了一路的坑过来的!)
第一步:安装VS2015
我安装的是这个版本:cn_visual_studio_professional_2015_x86_x64_dvd_6846645.iso
VS2015也可以选择社区版本,安装选项中可以只选择更新以及C++库
安装VS2015的过程比较漫长,大概需要两个小时左右,需要耐心等待~
第二部:安装CUDA9.0
CUDA安装时这个安装过程最为关键的一部,直接影响到能否成功安装!如果安装失败了,需要彻底的卸载,然后进行重装,如果卸载的不干净,cuda会大概率安装失败,进而导致整个过程安装失败,网上有些教程是可以通过VS编译示例程序来检测是否安装成功的。
这里笔者选择的是这个版本:cudnn-9.0-windows10-x64-v7.zip
这些是可以通过在官网填写问卷后下载的,也可以私聊作者,通过百度云进行分享~
这里Driver components是可以不用安装,不然可能会因为与电脑中自带的驱动冲突,导致安装失败。
如果安装成功的话,就可以进入下一步了(网上有很多方法判断是否编译成功,稍微有些繁琐,不测试的话,也不影响后续的操作,如果后面发现安装失败的话,十有八九是因为cuda安装失败了,这里就需要好好验证一下是否是这里出问题了,然后需要卸载cuda重新安装)。
第三步:安装cudnn
同样的在官网下载对应版本的cudnn,我这里选择的是:cudnn-9.0-windows10-x64-v7(值得注意的是,cudnn需要与cuda的版本相对应)
然后在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0路径下进行一一覆盖。
第四部:安装anaconda
笔者这里选择的是:Anaconda3-5.1.0-Windows-x86_64
个人觉得anaconda版本的选择对安装成功与否关系应该不大。anaconda安装过程很简单。需要注意的是:
需要把环境变量的勾给点上,避免后续的麻烦。
还有一点曾经看过一篇博客说安装路径最好放在c盘,也就是最好不要更改默认路径,避免出现问题。
第五步:安装TensorFlow
a)win+r 输入cmd打开命令行,输入 conda create -n tensorflow python=3.6注意这里的TensorFlow是你命名的虚拟环境名称,你可以取任意的名字,只需要记住就行。安装好后,你可以通过输入python,查看当前版本的python的版本。
b)win+r 输入cmd打开命令行,输入 activate tensorflow 激活刚才创建的虚拟环境。
输入pip install tensorflow-gpu==1.8 -i https://pypi.tuna.tsinghua.edu.cn/simple
等待安装,这里需要注意的是==1.8不可以忽略,因为少了这个,会默认安装最新版本的tensorflow,而最新版本的可能会与cuda不匹配,导致最终安装失败,所以,这里希望还是选择1.8版本,当然其他版本也都可以尝试一下,可能也行的通
c)测试安装是否成功
首先win+r 输入cmd打开命令行,输入activate tensorflow激活环境,输入python显示当前版本号,接着输入import tensorflow as tf,如果没有报错,那么恭喜你~安装成功啦~
第六步:安装pycharm
最后一步可以根据个人喜好选择安装IDE,这里介绍一下pycharm,pycharm社区版是免费的,我这里选择的是:pycharm-professional-2018.3.3专业版(感觉没多大的区别,专业版需要激活,需要修改hosts文件,可以自行百度)。
安装成功后,编译器选择刚才虚拟环境安装的位置,我这里h1024是我电脑名称。
OK!大家愉快地在pycharm下使用tensorflow吧~