window 10 下tensorflow 1.6 GPU版本的安装方法

    前些日子在 win10 下安装TensorFlow 1.2 GPU版本,但是目前在跑一些官方demo的时候发现1.2版本的不支持了。由于电脑自带GTX1051 Ti 想充分利用资源就安装了一个前几天最新发布的1.6 GPU版本TensorFlow。

    首先,你得可以翻墙,不然后面安装下载要用的东西会很慢。。。嗯,翻墙我就不多说了吧。

   既然,能翻墙,那么我们就开始吧。

    1,下载NVIDIA CUDA Toolkit:

        连接传送门:https://developer.nvidia.com/cuda-90-download-archive

    选择安装版本:嗯,看下图。。。。。我选的是深绿色的选项哦~~~~~~


    安装类型选择local 下载到本地

    然后弹出下载的东东,见图

    刚好是第一个,看到了吧?1.4G 挺大的,网速不给力的就等等吧,然后不能翻墙,基本就卡在这里了。。

    (如果是其他系统,自己选择吧)

    下载完NVIDIA CUDA Toolkit就安装它,(最好路径啥的都用英文,这是个好习惯,能省很多bug)

    安装过程稍微有点久。。。。。。。估计十几分钟?????

    如果以前下载并安装过较早版本的 NVIDIA CUDA Toolkit 不用管直接安装,安装过程中会自动卸载忽略旧版本。

    安装完过后可能会叫你重启电脑,你就重启吧。。。。。。。嗯~~~~~~

    2,安装cuDNN 7.x

    连接传送门:https://developer.nvidia.com/cudnn

    往下 滑 有个绿色的大 Download 按钮点进去。(这需要成员才能登录,注册个账号就行了。。)

    这里有一堆版本供你选择 ,1.6GPU 版本的TensorFlow 就选这个了 (见下图):

    然后你在选择你系统对应的版本即可(见下图)

    版本号格式:cudnn-9.0-windows[your version]-x64-v7.zip

    下载完了继续:

    解压你下载好的文件,等到一个cuda文件夹,进入这个cuda文件夹把这个文件夹里的东西都复制,然后拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0文件夹下面。

    看看cuda文件夹里面的东西,见下图:(所有拷贝额~~~)



    3,下载whl文件

    连接传送门:https://pypi.python.org/pypi/tensorflow-gpu

    找到对应的版本,我python版本是3.6的,所以我选择了:

    tensorflow_gpu-1.6.0-cp36-cp36m-win_amd64.whl (md5) 下图最后一个

    嗯,到这里就快完成了。

    运行命令:pip install [whl file path]  路径我就不多说了吧?

===================分割线 ====================

    到这里,就恭喜你完成了安装了。。检查安装是够成功。。

运行一下命令:

    import TensorFlow as tf

    tf.__vresion__         就可以看到你当前的版本了






 

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值