Tensorflow1.4安装gpu教程

以前安装的是tensorflow-cpu,但是随着运算量的提高产生了鼓捣tensorflow-gpu的想法。官网的教程看起来很简单,但是坑很多。本文主要是帮助大家绕过那些坑,成功安装可以运行的tensorflow-gpu。

  1. 官网安装教程链接
    官网关于Tensorflow-gpu的软件要求
  2. 这要求我们安装CUDA Toolkit 8.0和cuDNN v6.1,注意Tensorflow 1.4必须是CUDA Toolkit 8.0,但是官网默认下载CUDA Toolkit 9.1。
  3. 下载CUDA Toolkit 8.0 ,之后一直点默认的就可以安装完成
  4. 下载cuDNN v6.0,注意下载这个安装包需要注册并且填一堆问卷,下下来以后解压,然后直接拷到cuda路径对应的文件夹下面就行。具体操作是讲将下图对应三个文件夹里面的文件分别拷贝到cuda安装目录对应文件夹
    这里写图片描述

  5. cuda安装目录如下图所示
    这里写图片描述

  6. 如果现实如图所示就证明tensorflow-gpu安装成功了
    这里写图片描述

如果不明白的话,大家可以参考点击

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值