初用Linux, 安装Ubuntu16.04+NVIDIA387+CUDA8.0+cudnn5.1+TensorFlow1.0.1

因为最近Deep Learning十分热门, 装一下TensorFlow学习一下. 本文主要介绍安装流程, 将自己遇到的问题说明出来, 并记录自己如何处理, 原理方面并没有能力解释. 由于本人之前从来没有用过Linux, 本文章恐有初级错误, 望见谅, 谢谢. (本文写于2017年3月17日)

 

为了能够利用GPU(NVIDIA GTX1080)运行TensorFlow, 根据调查需要按顺序安装以下内容:

Ubuntu, NVIDIA驱动, CUDA, cudnn, TensorFlow

 

安装Ubuntu16.04

Ubuntu16.04 可以通过官网下载, 并制作成U盘安装. 为了更方便的驱动显卡,直接安装Linux, 并不安装双系统或虚拟机. 安装完毕后, 由于Ubuntu源在国外, 为了更效率得更新, 采用中科大的镜像源.
具体操作为按Ctrl+Alt+T打开命令行, 然后输入:

sudo gedit /etc/apt/sources.list

第一次以管理员运行, 需要输入管理员密码. 然后将内容替换为:

# 默认注释了源码镜像以提高 apt update 速度，如有需要可自行取消注释
deb https://mirrors.ustc.edu.cn/ubuntu/ xenial main restricted universe multiverse
# deb-src https://mirrors.ustc.edu.cn/ubuntu/ xenial main main restricted universe multiverse
deb https://mirrors.ustc.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse
# deb-src https://mirrors.ustc.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse
deb https://mirrors.ustc.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse
# deb-src https://mirrors.ustc.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse
deb https://mirrors.ustc.edu.cn/ubuntu/ xenial-security main restricted universe multiverse
# deb-src https://mirrors.ustc.edu.cn/ubuntu/ xenial-security main restricted universe multiverse

# 预发布软件源，不建议启用
# deb https://mirrors.ustc.edu.cn/ubuntu/ xenial-proposed main restricted universe multiverse
# deb-src https://mirrors.ustc.edu.cn/ubuntu/ xenial-proposed main restricted universe multiverse

然后更新源和安装包:

sudo apt-get update
sudo apt-get upgrade

*安装界面选择取消选择从网络下载. 即便如此, 在之后的Retriving File中, 也有可能遇到进度条卡死的情况(我是校园网, 需要网络登录), 处理方法是拔掉网线, 待安装完毕后再将网线插回. 

 

安装NVIDIA378.13

安装NVIDIA最新显卡驱动, 从官网输入自己的显卡型号, 点击search, 提示下载显卡驱动为375.39(Release 2017.2.14), 然而在输入自己显卡型号的页面, 选择下方的"Beta and Older Drivers"的连接, 可以选择到378.13. 由于我在用375.39的时候, 之后通过Source方式安装TensorFlow中遇到问题(这里可以先不用管), 搜索得到推荐使用378的型号, 改用378.13的驱动. 将该文件移动到Home目录中, 改名为"NVIDIA.run"



 

具体安装方法为: Ctrl+Alt+F1, 进入命令行页面, 输入用户名和密码登录, 首先关闭X Server(我理解是图形界面):

sudo /etc/init.d/lightdm stop

然后运行NVIDIA.run, 进行安装:

sudo sh NVIDIA.run # 注意此时文件已在Home目录下

安装中, 会遇到"The distribution-provided pre-instrall script failed", 可以无视, 继续安装. 但会出现关于Neuveau的问题, 好在NVIDIA会帮你把Neuveau给加到blacklist中, 只需要同意这么做就好, 但此次安装会失败, 退出, 这时需要输入:

sudo update-initramfs -u

然后键入reboot, 重启计算机. 此时的桌面分辨率会降低, 按Ctrl+Alt+F1进入命令行, 管理员登入, 关闭X Server, 然后安装NVIDIA驱动, 过程中会说NVIDIA会把其他X Server给遮蔽掉(大概这么个意思?), 同意即可, 然后便顺利安装. 之后reboot重启即可.

* 如果安装完毕后, 在登录界面循环登录, 即输入密码, 又退回的话, 关闭掉主板的Secret Boot即可. 

 

安装CUDA8.0

官网选择Linux-x86_64-Ubuntu-16.04-Runfile(local), 然后将文件重命名CUDA.run并移动到Home根目录中, 按Ctrl+Alt+T打开命令行, 运行:

sudo sh CUDA.run

一路空格搞到底部, 键入accept同意安装, 之中有些地方需要选择, 除了在询问是否安装:NVIDIA驱动375.xx选否(已经安装了378.13), 其他的都同意并使用默认路径, 即可以完成安装. 最后加入环境变量:

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

再通过以下命令让环境变量生效:

source ~/.bashrc

* 如果在安装CUDA时遇到空间不足的问题, 可以在运行CUDA.run时加入:

sudo sh CUDA.run --tmpdir=/opt/temp/

我个人没有遇到这个问题, 同时如果加入后面的代码, 反而会无法运行.

安装cudnn5.1

下载好的压缩包重命名为cudnn.tgz并移动到Home根目录中, 然后运行命令解压:

tar -zxvf cudnn.tgz

会得到五个文件, 将文件拷贝到对应的CUDA目录下即可:

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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

 

安装TensorFlow

根据官网, Ubuntu上可以有5种方式, 分别是利用: Virtualenv, "native" pip, Docker, Anaconda, 以及从Source安装. 由于我在通过Source和Virtualenv中最后一步都遇到问题,
使用Anaconda成功安装. 

1
a8eb
. 安装Anaconda, 具体可参见Anaconda官网

2. 创建conda环境, 用命令:

source create -n tensorflow

激活conda环境, 此时您的命令行前多了(tensorflow):

这里是你关于您计算机的名字 $ source activate tensorflow
(tensorflow) 一些关于您计算机的名字 $

安装TensorFlow在conda环境中, 最后的连接是TensorFlow Python Package, 根据需要来选择, 由于我使用GPU, Linux系统, Python2.7, 使用如下命令

(tensorflow)一些关于您计算机的名字$ pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl[/code] 
安装完毕后, 尝试运行

(tensorflow)一些关于您计算机的名字$ python >>> import tensorflow as tf


如果可以运行, 恭喜安装成功. 之后可以通过如下命令来退出conda环境

(tensorflow)一些关于您计算机的名字$ source deactivate tensorflow


* 如遇到问题类似 ImportError: libcudart.so.Version: cannot open shared object file, 可以通过下面命令查看一下自己的环境变量, PATH和LD_LIBRARY_PATH是否都包含全了

export


我最终的PATH包含如下路径:

declare -x PATH="/home/你的用户名/anaconda2/envs/tensorflow/bin:/home/你的用户名/anaconda2/bin:/usr/local/cuda-8.0/bin:usr/local/cuda/bin:(一些其他的路径)"


如果有缺少的可以尝试使用export PATH="$PATH:缺少的路径"补充, 记得最后source ~/.bashrc

 (事实上我在用Source和Virtualenv遇到就是这个问题, 可惜当时并不知道怎么解决)