ubuntu16.04下安装tensorflow(gpu加速)----详细图文教程

蓝色荣誉 蓝色荣誉     2022-12-11     153

关键词:

写在前面

一些废话

接触深度学习已经有一段时间,之前一直在windows下使用Theano,但是发现Theano天书般的源码真是头大,在看到tensorflow中文教程后,发现它竟然逻辑清晰,教程丰富,实在是居家旅行必备良药啊![偷笑][偷笑][偷笑]

所以决定利用国庆假期学习ubuntu和TensorFlow的安装,结果入坑无数,同时搞坏了一块1T硬盘(花了450大洋啊,心在滴血…)。初步估算,整个安装过程耗时10.2-10.11,共10天(除去玩的几天,应该有1周时间在配置),安装windows 5+次,安装ubuntu 20+次,请叫我装机小能手。。。这主要原因是:我是小白,我的笔记本太奇葩,网上教程太混乱。

为了减少以后各位的入坑时间,把更多的精力放在Deep Learning的学习上,也为了纪念我这么多天的辛苦,所以献上超级无敌吊炸天的终极TensorFlow安装教程!

配置

硬件:Thunderbot 911笔记本,CPU:i7,GPU :GeForce GTX 960m, 8G内存,120G SSD+1T 机械硬盘。
软件: ubuntu16.04+cuda8.0+cudnn v5+tensorflow 0.11


1. 下载

1.1 系统镜像

由于我尝试了ubuntu14.04,安装Nvidia驱动之后,会出现循环登录的问题,并始终无法找到有效的解决途径,所以只能选择ubuntu16.04了。
镜像地址 https://www.ubuntu.com/download/alternative-downloads

1.2 CUDA 8.0

https://developer.nvidia.com/cuda-downloads(下载地址)
说明:
(1)在NVIDIA的CUDA下载页面下,选择要使用的CUDA版本进行下载。
(2)我们这里使用CUDA8.0(页面有提示GTX1070、GTX1080支持8.0版本),学员如果没有使用以上两个版本的GPU,可以下载CUDA7.5。DOWNLOAD(下载)。
(3)下载需要注册。
(4)图解选择

1.3 cuDNN v5

下载地址: https://developer.nvidia.com/cudnn(需要登录)
说明:
(1)下载需要填写一个调查问卷,就三个选项,建议认真填写,毕竟人家免费给咱使用。
(2)填写完毕点击 I Agree To 前面的小方框,出现如下:

1.4 Tensorflow 0.11

tensorflow github上面提到 4 种安装方式,本教程使用 第四种 源码安装
Virtualenv installation
Anaconda installation
Docker installation
Installing from sources
https://github.com/tensorflow/tensorflow(下载地址)
说明:
(1)打开下载页面,往下翻,直到下图这个位置:


(2) 点击Python 2开始下载。

最后,将1.2-1.4中下载文件全部存放至自己的移动硬盘/U盘内,等待安装时候使用。

2. 安装ubuntu16.04 LTS 系统

安装Ubuntu16.04:
http://jingyan.baidu.com/article/eb9f7b6d8536a8869364e813.html
说明:
(1)我们直接安装的英文原版系统,语言也是选择英文的。
(2)上述链接在–第三步:安装类型上选择的是–自定义。我们选择的是–清除整个磁盘并且安装,如果你有Windows系统,还会提示安装Ubuntu16.04与Windows并存模式。这个自行选择,切记!这个地方谨慎选择。
(3)感谢百度经验上传者!

3. 安装NVIDIA驱动

打开terminal输入以下指令:

sudo apt-get update

然后在系统设置->软件更新->附加驱动->选择nvidia最新驱动(361)->应用更改

3. cuda 8.0

3.1 安装cuda

在cuda所在目录打开terminal依次输入以下指令:

cd  /home/***(自己的用户名)/Desktop/###(这个命令意思是找到刚刚我们用U盘传过来的文件)
sudo dpkg -i cuda-repo-ubuntu1604-8-0-rc_8.0.27-1_amd64​.deb
sudo apt-get update
sudo apt-get install cuda​

3.2 gcc降版本

ubuntu的gcc编译器是5.4.0,然而cuda8.0不支持5.0以上的编译器,因此需要降级,把编译器版本降到4.9:
在terminal中执行:

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++

3. 安装cuDNN

打开terminal依次输入以下指令:

cd  /home/***(自己的用户名)/Desktop/###(这个命令意思是找到刚刚我们用U盘传过来的文件)
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*

4. 安装其他依赖

4.1 配置环境变量


按照上图的教程,在terminal中输入以下命令:

sudo gedit ~/.bash_profile #打开.bash_profile

然后在打开的文本末尾加入:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

继续在terminal中输入:

source ~/.bash_profile #使更改的环境变量生效

当然,也有其他教程在文件~/.bashrc文件中写入的,方法与上面的类似。如果在后面配置./config文件出现问题时,可以实现这个方法。

4.2 安装其他库

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/get_started/os_setup.md
我们是在github的Tensorflow官方网页上,根据提示安装,地址如上。
按步骤截图如下

在terminal中输入以下命令:

sudo apt-get install python-pip python-dev 

4. 安装Bazel

4.1 安装Bazel依赖

由于本教程使用tensorflow源码编译/安装,所以需要使用 bazel build。
链接:https://www.bazel.io/versions/master/docs/install.html

在terminal中依次输入以下1-7的命令

4.2 安装Bazel

之后回到之前的Tensorflow安装教程页面:https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/get_started/os_setup.md

点击链接: installer for your system,跳转到Bazel的下载页面:

下载bazel-0.3.2-installer-linux-x86_64.sh到桌面,然后在terminal中输入以下命令

cd  /home/***(自己的用户名)/Desktop/###(这个命令意思是找到刚刚我们用U盘传过来的文件)
chmod +x PATH_TO_INSTALL.SH #对.sh文件授权
./PATH_TO_INSTALL.SH --user #运行.sh文件

4.3 安装第三方库

在terminal中输入以下命令

sudo apt-get install python-numpy swig python-dev python-wheel #安装第三方库
sudo apt-get install git
git clone git://github.com/numpy/numpy.git numpy 

5. 安装tensorflow

5.1 下载tensorflow

在terminal中输入以下命令

git clone https://github.com/tensorflow/tensorflow

特别注意,我使用的是tensorflow 0.11版本,该版本要求cuda 7.5 以上,cuDNN v5。
默认下载目录是在/home下

5.2 配置tensorflow

还是刚刚的网址
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/get_started/os_setup.md

在terminal中输入以下命令:

cd ~/tensorflow #切换到tensorflow文件夹
./configure #执行configure文件

然后按照下图选项进行操作:

5.3 创建pip


在terminal中输入以下命令:

bazel build -c opt //tensorflow/tools/pip_package:build_pip_package
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
sudo pip install /home/***(你自己的用户名)/Desktop/tensorflow-0.10.0-cp2-none-any.whl

5.4 设置tensorflow环境

bazel build -c opt //tensorflow/tools/pip_package:build_pip_package
 # To build with GPU support:
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
mkdir _python_build
cd _python_build
ln -s ../bazel-bin/tensorflow/tools/pip_package/build_pip_package.runfiles/org_tensorflow/* .
ln -s ../tensorflow/tools/pip_package/* .
python setup.py develop

这样就大功告成啦~!!!

6. 测试tensorflow

这里进行测试,如果你能跟我看到同样的画面,那恭喜你成功配置GPU版的tensorflow啦!


跑这个例子,会出现很多提示,如果你在运行过程中发现自己的显卡型号,并提示成功调用cuda库,并每次step小于100ms,说明成功,否则就检查下哪里出现问题吧~
下面就尽情调戏tensorflow啦!
这里给出很有意思的教程链接:http://m.blog.csdn.net/article/details?hmsr=toutiao.io&id=52658965&utm_medium=toutiao.io&utm_source=toutiao.io
用tensorflow实现梵高作画。

7. 常见问题

7.1循环登录

在ubuntu14.04安装N卡驱动后,会出现无法显示登录界面或者循环登录的问题。这主要是显卡不兼容,具体解决思路可以参考google上的解决方案,关键词 ubuntu login loop。
经过测试,网上的教程对我都不适用,无奈转向ubuntu16.04

7.2 缺少第三方库

因为这个教程是我安装成功之后写的,其中难免遗忘某些库的安装,例如git、pip这些库,安装过程很简单,具体可以google。

7.3 tensorflow配置问题

在执行./configure 或者设置tensorflow环境时,如果出现无法找到某个库的路径,那么检查是否正确的设置了cuda的环境变量,具体参考 4.1节。

7.4 cuda8.0不支持gcc 5.3以上版本


这个问题可以通过对gcc降版本解决。相关连接 http://m.blog.csdn.net/article/details?id=51999566

7.5 测试tensorflow时出现IOError

在测试tensorflow中,执行

python convolutional.py

出现 IOError错误,这是由于convolutional.py中需要从网上下载MNIST数据库。如果出现错误,那么重新执行python convolutional.py命令,或者手动从网站下载数据库并放在相应文件夹就好啦。

8. 经验与总结

  1. google是最好的老师!
  2. 感谢七月在线团队的无私帮助: qq群:472899334
  3. 失败是成功之母,经过这么多次尝试,以后的配置应该都不是问题啦
  4. 欢迎联系我的QQ: 3062984605
  5. 欢迎留言补充或讨论

9. 参考文献

[1] http://blog.csdn.net/u010789558/article/details/51867648
[2] http://textminingonline.com/dive-into-tensorflow-part-iii-gtx-1080-ubuntu16-04-cuda8-0-cudnn5-0-tensorflow
[3] http://m.blog.csdn.net/article/details?id=52658965
[4] https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/get_started/os_setup.md#installing-from-sources
[5] http://www.tensorfly.cn/tfdoc/get_started/os_setup.html
[6] http://ramhiser.com/2016/01/05/installing-tensorflow-on-an-aws-ec2-instance-with-gpu-support/
[7]http://blog.csdn.net/u012436149/article/details/52554176
[8] http://m.blog.csdn.net/article/details?id=51999566


利用pip安装方法 ##(版权归属: QQ 1395569872)

Ubuntu16.04从U盘安装纯净单系统

Ubuntu16.04安装NVIDIA显卡官方驱动
1.点桌面左上角搜索本机程序的图标,找到“附加驱动”
2.在“附加驱动”里,系统会自动搜索N卡驱动,列表里会提供对应你显卡的最新版官方驱动。例如我的显卡是GT730,选择第一项361.42就可以了。
3.最后点“应用更改”,等待安装完毕即可。

安装CUDA【Debian安装】
1、下载安装
进入下载文件所在目录,执行下列命令:

$ sudo dpkg --install cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

安装cuDNN
1、下载安装Cudnn v5.1(https://developer.nvidia.com/cudnn
进入下载文件所在目录,执行下列命令:

$ tar xvzf cudnn-8.0-linux-x64-v5.1.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*

2、配置环境变量:
在terminal根目录中输入以下命令:

$ sudo gedit ~/.bash_profile

然后在打开的文本末尾加入:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

继续在terminal中输入:

$ source ~/.bash_profile 

安装pip

$ sudo apt-get install python-pip python-dev
$ sudo apt-get install python-numpy swig python-dev python-wheel

安装TensorFlow

# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.

$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

$ sudo -H pip install --upgrade $TF_BINARY_URL

ubuntu16.04下安装cuda,cudnn及tensorflow-gpu版本过程(代码片段)

...3;CUDA8.0深度学习主机环境配置:Ubuntu16.04+GeForceGTX1080+TensorFlow不过在实际运行的过程中,有一定的不同之处,随着时间的推移,一些组件已经可以更方便的安装, 查看详情

ubuntu16.04下安装cuda,cudnn及tensorflow-gpu版本过程(代码片段)

...3;CUDA8.0深度学习主机环境配置:Ubuntu16.04+GeForceGTX1080+TensorFlow不过在实际运行的过程中,有一定的不同之处,随着时间的推移,一些组件已经可以更方便的安装,不再需要自己编译了。一些流程也有所更改。因... 查看详情

ubuntu16.04安装tensorflow(cpu版本)详细可行步骤

参考技术A2、Tensorflowcpu版本的安装1、是pip3(Python3.5版本),不是pip(python2.7版本);2、输入命令行时,空格会导致输入不正确,得注意;3、测试安装成功1、命令行输入:python2、继续输入:重装Anaconda时,关于卸载Anaconda3的方法... 查看详情

求助tensorflow下遇到cudacomputecapability问题

...装教程可以参考:Ubuntu16.04上gtx1080的cuda安装July172016目前tensorflow是一个非常流行的深度学习计算框架,常规硬件及系统的安装方法官方的doc已经说的很清楚了,但是因为系统是ubuntu16.04,显卡是GTX1080,所以不可避免的要折腾起来。... 查看详情

Ubuntu 16.04 上 TensorFlow 的 NVIDIA cuDNN 版本类型 [关闭]

】Ubuntu16.04上TensorFlow的NVIDIAcuDNN版本类型[关闭]【英文标题】:WhichNVIDIAcuDNNreleasetypeforTensorFlowonUbuntu16.04[closed]【发布时间】:2018-07-2421:01:37【问题描述】:根据TensorFlow1.5installationinstructionsforUbuntu16.04,您需要安装cuDNN7.0,但他们... 查看详情

怎么在ubuntu16.04上安装tensorflow

...anttocontinue?[Y/n]y$sudopipinstall--upgradehttps://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.8.0-cp27-none-linux_x86_64.whl$wgethttps://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.8.0-cp27-none-linux_x86_64.whl$sudopipinstalltensorflow-0.8.0-cp27-none-linux_x86_64.whl此... 查看详情

ubuntu16安装tensorflow及jupyternotebook(代码片段)

Ubuntu16安装TensorFlow及Jupyternotebook文章目录Ubuntu16安装TensorFlow及Jupyternotebook1.软件版本2.安装过程3.测试本文主要讲述Windows上VMware虚拟机中Ubuntu16.04(最小化安装版,没有安装界面环境)安装TensorFlow及Jupyternotebook的过程1.... 查看详情

ubuntu16安装tensorflow及jupyternotebook(代码片段)

Ubuntu16安装TensorFlow及Jupyternotebook文章目录Ubuntu16安装TensorFlow及Jupyternotebook1.软件版本2.安装过程3.测试本文主要讲述Windows上VMware虚拟机中Ubuntu16.04(最小化安装版,没有安装界面环境)安装TensorFlow及Jupyternotebook的过程1.... 查看详情

阿里云ubuntu16.04系统下安装nginx

阿里云Ubuntu16.04系统下安装Nginx 查看详情

ubuntu16.04安装nvidia驱动cuda,cudnn和tensorflow-gpu(代码片段)

本文个人博客地址:点击查看之前有在阿里云GPU服务器上弄过:点击查看,这里从装Nvidia开始一、安装Nvidia驱动1.1查找需要安装的Nvidia版本1.1.1官网官网上查找:https://www.nvidia.com/Download/index.aspx?lang=en-us这里是Ge... 查看详情

ubuntu16.04下安装petalinux(代码片段)

ubuntu16.04下安装petalinux参考https://www.cnblogs.com/vacajk/p/6146146.html安装环境:  1.ubuntu16.04  2.petalinux-v2015.4  3.目标板:zynq7010SOC安装步骤:  1.在home目录下新建两个文件夹:petalinux和petalinux_installer两个文件夹,将安装包petali 查看详情

在ubuntu16.04下安装nodejs

源安装:1.curl-sLhttps://deb.nodesource.com/setup_5.x|sudo-Ebash- 2.sudoapt-getinstall-ynodejs使用命令安装看到此图,安装成功!使用node-v查看弄的js版本号   查看详情

ubuntu16.04下安装mysql详解(代码片段)

Ubuntu16.04下安装MySQL详解分别依次输入以下3个命令:sudoapt-getinstallmysql-serversudoaptinstallmysql-clientsudoaptinstalllibmysqlclient-dev安装成功后可以通过下面的命令测试是否安装成功:sudonetstat-tap|grepmysql 出现如下信息证明安装成功:>&... 查看详情

普通电脑pc怎样跑tensorflow的gpu模式

...装教程可以参考:Ubuntu16.04上gtx1080的cuda安装July172016目前tensorflow是一个非常流行的深度学习计算框架,常规硬件及系统的安装方法官方的doc已经说的很清楚了,但是因为系统是ubuntu16.04,显卡是GTX1080,所以不可避免的要折腾起来。... 查看详情

ubuntu16.04下怎么安装flashplayer

我用的是firefox.直接aptinstalladobe-flashplugin即可.本文出自“12208412”博客,请务必保留此出处http://12218412.blog.51cto.com/12208412/1886250 查看详情

ubuntu16.04下sublimetext3安装和配置

ubuntu16.04下sublimetext3安装和配置2018年04月20日10:31:08zhengqijun_阅读数:14821.安装方法1)使用ppa安装sudoadd-apt-repositoryppa:webupd8team/sublime-text-3sudoapt-getupdatesudoapt-getinstallsublime-text-installer卸载sublime 查看详情

ubuntu下自动安装tensorflow脚本

(1)前提是你得安装好驱动和cuda安装驱动和cuda很简单,到以下位置选择ubuntu的cuda包即可https://developer.nvidia.com/cuda-toolkit然后sudodpkg-i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.debsudoapt-getupdatesud 查看详情

ubuntu16.04安装nvidia驱动cuda,cudnn和tensorflow-gpu(代码片段)

本文个人博客地址:点击查看之前有在阿里云GPU服务器上弄过:点击查看,这里从装Nvidia开始一、安装Nvidia驱动1.1查找需要安装的Nvidia版本1.1.1官网官网上查找:https://www.nvidia.com/Download/index.aspx?lang=en-us这里是Ge... 查看详情