Win10 64位系统 TensorFlow-Gpu安装(CUDA9.0+CUDNNv7+VS2015+1050Ti+tensorflow-gpu1.8)

笔者亲试很多次,在自己笔记本(联想拯救者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吧~

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值