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安装成功了
    这里写图片描述

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