热门标签 | HotTags
当前位置:  开发笔记 > 人工智能 > 正文

Tensorflow1.4安装gpu教程

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

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

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


推荐阅读
author-avatar
书友10689978
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有