专栏首页磐创AI技术团队的专栏Ubuntu 安装 tensorflow-gpu 1.4 +CUDA 8.0 +cuDNN详细教程

Ubuntu 安装 tensorflow-gpu 1.4 +CUDA 8.0 +cuDNN详细教程

作者 | fendouai

编辑 | 磐石

出品 | 磐创AI技术团队

【磐创AI导读】:本文详细介绍了tensorflow-gpu在Ubuntu下的安装步骤。欢迎大家点击上方蓝字关注我们的公众号:磐创AI

硬件环境:NVIDIA GTX 980 Ti

系统环境:Ubuntu 16.04 64位

一.安装 NVIDIA驱动

1. 关闭 Secure Boot

具体如何禁用 BIOS 中的 Secure Boot 要根据主板的情况。

以华硕主板的禁用方法为例:

首先进入 BIOS,然后选择 Boot ,进入 Secure Boot 界面:

确定 "OS Type" 是 "Windows UEFI"

进入 "Key Management"

选择 "Clear Secure Boot keys"

在你清除 " Secure Boot keys" 之后,你将会有 "Install default Secure Boot keys" 选项来恢复默认的 keys 。

在你清除 " Secure Boot keys" 之后, Secure Boot 会被自动关闭,你现在可以设置 "OS Type" 为 "Other OS"。

设置完成之后的效果:

2.禁用 nouveau driver

控制台输入命令,创建一个文件通过命令

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

并添加如下内容:

blacklist nouveau

options nouveau modeset=0

再更新一下

sudo update-initramfs -u

修改之后需要重启系统。确认下nouveau是已经禁用,可以使用命令:

lsmod | grep nouveau

3.在字符界面下安装驱动

首先添加ppa库,然后通过ppa安装显卡驱动,使用以下命令添加:

sudo add-apt-repository ppa:graphics-drivers/ppa

关闭图形化环境

首先进入 Ubuntu 系统字符界面,使用组合键 ALT+CTRL+F1 进入字符界面。

为了确保驱动能够正常安装,我们需要暂时关闭x-window服务(图形环境),在文本模式下输入命令进行关闭:

sudo service lightdm stop

安装驱动

首先通过以下命令来查看 NVIDIA VGA card model

sudo lshw -numeric -C display

可以使用命令:

ubuntu-drivers devices

来查看可以使用的驱动,如图:

输入以下命令,直接安装驱动:

sudo apt-get update && sudo apt-get install nvidia-384

进入系统,我们要看刚刚的显卡驱动是否已经正确安装成功,通过下面命令查看:

nvidia-smi

如图所示显卡驱动已经正确安装,显卡的型号是 GTX 980 Ti。

二.安装 CUDA

1.确定和 TensorFlow 对应的 CUDA 版本

TensorFlow GitHub 页面查看依赖的版本:

https://github.com/tensorflow/tensorflow/releases

2. 下载和安装

因为 CUDA 8 兼容的gcc 版本是 4.9 ,但是ubuntu 16.04 默认的是 gcc 5,所以需要gcc降版本。

sudo apt-get install g++-4.9

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10

sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20

sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10

sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30

sudo update-alternatives --set cc /usr/bin/gcc

sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30

sudo update-alternatives --set c++ /usr/bin/g++

下载地址:

https://developer.nvidia.com/cuda-downloads

因为上一步已经安装了显卡驱动,所以这里要选 no 。

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48?

(y)es/(n)o/(q)uit: n

3.设置环境变量

配置CUDA环境变量

export PATH="$PATH:/usr/local/cuda-8.0/bin"

export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64"

三.安装cuDNN

1.确定版本

TensorFlow GitHub 页面查看依赖的版本:

https://github.com/tensorflow/tensorflow/releases

2.下载

下载地址:

https://developer.nvidia.com/cudnn

3.安装

tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

四.安装 tensorflow-gpu

直接安装

pip install tensorflow-gpu

指定版本安装:

例如,要安装 tensorflow-gpu 1.4

pip install tensorflow-gpu==1.4

安装完成

参考教程:

https://blog.csdn.net/roach_zfq/article/details/53883976

https://www.cnblogs.com/xuliangxing/p/7569946.html

https://www.linuxidc.com/Linux/2016-11/137561.htm

https://blog.csdn.net/qq_30520759/article/details/78947034

你也许还想看:

● 一文彻底搞懂BP算法:原理推导+数据演示+项目实战(上篇)

● TensorFlow + Keras 实战 YOLO v3 目标检测图文并茂教程(文末有惊喜)

● 入门 | Tensorflow实战讲解神经网络搭建详细过程


Tips:欢迎大家点击最下方二维码关注我们的公众号,点击干货资源专栏或发送关键字“资源”获取更多资源推荐。关注我们的历史文章,一起畅游在深度学习的世界中。我们期待你的留言和投稿,共建交流平台。来稿请寄:voice1235@163.com

本文分享自微信公众号 - 磐创AI(xunixs),作者:fendouai

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2018-05-25

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 为何Keras中的CNN是有问题的,如何修复它们?

    上个星期我做了一些实验,用了在 CIFAR10 数据集上训练的 VGG16。我需要从零开始训练模型,所以没有使用在 ImageNet 上预训练的版本。

    磐创AI
  • 一文上手Tensorflow2.0(四)

    【磐创AI导读】:本系列文章介绍了与tensorflow的相关知识,包括其介绍、安装及使用等。本篇文章是本系列文章的最后一篇。查看上篇:一文上手Tensorfl...

    磐创AI
  • 边做边思考,谷歌大脑提出并发RL算法,机械臂抓取速度提高一倍!

    该研究在如下情况中研究强化学习:在受控系统随着时间演变的过程中同时对动作进行采样。换句话说,所研究的机器人必须在执行着上一个动作的同时考虑下一个动作。就如同人或...

    磐创AI
  • [Windows10]安装用友U870遇到的2个问题

    1.安装.NET framework 3.5 安装报错 0x800F0954 处理方法: 打开注册表-->win+r 输入regedit 找到HKE...

    贰叁壹小窝
  • Ubuntu中配置TensorFlow使用环境的方法

    TensorFlow™是一个基于数据流编程(dataflow programming)的符号数学系统,被广泛应用于各类机器学习(machine learning...

    砸漏
  • 二分查找变种

    该算法有很多版本,这里给出java中实现比较好的一种方式。其中,>>>为无符号右移。

    xiaoxi666
  • python django 上传图片

    reqfile= request.FILES['p_w_picpath']     

    py3study
  • 《Go 语言程序设计》读书笔记 (一)基础类型和复合类型

    最近在读《Go 语言程序设计》这本书想通过看书巩固一下自己的基础知识,把已经积累的点通过看书学习再编织成一个网,这样看别人写的优秀代码时才能更好理解。当初工作中...

    KevinYan
  • OCP-052考试题库汇总(41)-CUUG内部解答版

    A)every update statement executed by scott that updates scott.emp

    用户5892232
  • 给小白的论文写作方法!实用率99%!

    身为80后的入门级“青椒”,经日摸索挣扎于学术研究的各个环节,写学术论文就是其中的典型一环。虽然在别人“直观”看来仅仅是一串统计数字,但写学术论文与做研究本身一...

    昱良

扫码关注云+社区

领取腾讯云代金券