当前位置:   article > 正文

Tensorflow1.4安装gpu教程_tensorflow-gpu 1.4找不到安装

tensorflow-gpu 1.4找不到安装

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

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

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号