[框架安装趟雷指南]Ubuntu+1060+cuda+cudnn+Keras+TH+TF+MXnet(转自知乎) 附带装mxnet0.9.5版本

原地址:https://zhuanlan.zhihu.com/p/23480983
写的很好，把他搬运过来 我的MXnet就是看这个装的，希望能帮助到大家
写这篇文章的问题是为了回报社会，同时纪念长达一周的地狱般的安装（折腾）过程。

本文不一定是最佳安装方法，也不一定适合每一个人，主要目的是帮助大家避坑。

已经更到安装完毕keras+th+tf并开启gpu支持，如果发现有任何错误请务必指正。

感谢一切在安装过程中给予我帮助的文档作者、群友、同行，感谢清华大学开源软件镜像站，感谢x雷。

重要的话写在前头：

1：不要尝试Windows安装！不要尝试Windows安装！不要尝试Windows安装！成功安装只属于不可复制的小概率事件！不要想着工作娱乐用Windows，在Windows上搞深度学习会方便！呵呵，等着系统完蛋工作娱乐一起毁吧。>.<

2：网上文档众多，不少过于老旧或者不适合你的环境，注意避坑！小trick：在用100度搜索时，用筛选功能筛进一周或者一个月的，选择环境硬件与自己相近的。

正文：

我的配置：

华硕主板，双路e5，GTX1060，240gINTEl固态装win，120gINTEL固态装Ubuntu，1T机械放东西。

最终系统及重要环境：Ubuntu16.04(sever)，cuda8.0.44，cudnn5.1

一.安装ubuntu（最大的坑！已有ubuntu跳过）

用N卡装ubuntu是一件很坑的事，安装界面都可能进不去。这是最大的坑，过去了以后其他的其实都是小坑。

避坑方法：安装sever版再安装Desktop！！！！！！！！！

sever 版优点是无图形界面，可以顺利安装并进入系统（命令行），至少能让计算机活过来。N卡福音。

缺点也是无图形界面，真的不习惯。

安装过程：

1.官网下载：Ubuntu Server

2启动盘制作

过程参照这一篇教程：Ubuntu 16.04 U盘安装图文教程_Linux教程_Linux公社-Linux系统门户网站

*注意，选择镜像类型时要选择Ubuntu sever installer ，这个和教程有区别。

3.备份！

4.插u盘重启进bios，设置u盘启动。保存退出（重启）

5.U盘启动成功，进入grub

安装过程基本上没坑，不会在那个界面截图，过程参考：Ubuntu Server 13.10 安装配置图解教程

*分区的时候需要注意，最好准备一块空硬盘单独安装ubuntu，这个和文档不一样

选择手动》》选择要安装的硬盘的名称那行》》建立空白分区列表》》选择空白分区》》自动分区》》确认写入

最后别忘了拔u盘，修改BIOS

6.安装完成，应该是黑屏。。。。

不要慌，sever就这样，默认进入黑屏！按ctrl+shift+f1进入命令行，输入用户名和密码！

升级和配置基本环境

sudo apt update
sudo apt upgrade
sudo apt install -y python-dev python-pip python-nose gcc g++ git gfortran vim
sudo apt install -y libopenblas-dev liblapack-dev libatlas-base-dev
二、配置n卡

第二大坑，根据另外一个显卡是集显或者核心有不同。

安装流程:

安装desktop>>安装驱动>>重启进desktop

sudo apt-get update
sudo apt-get install ubuntu-desktop
桌面安装完毕，如果进的去，就在附加驱动里看一看N卡用什么驱动，最好换367以上。

不过有可能进不去，切到命令行，安装显卡驱动。

方法1:

wget:驱动网址

sudo service lightdm stop
sudo sh 文件名

sudo service lightdm restart
这里网址我是手打，run文件的

方法2:

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-367
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev
sudo service lightdm restart
重启，应该能成功进入桌面系统了

如果闪退，就尝试重启，再不行就再装一次驱动

三、安装cuda

在NVIDIA下载cuda8的runfile local版，注意8.0.44已经解决了gcc版本这个大坑！

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

*小trick，可以在Windows下用x雷下载，速度爆快！

拷到home目录

sudo sh cuda_8.0.44_linux.run
当它询问是否要安装cuda包里带的驱动时(我记得是367.xx)，一般教程都说否，因为他们装cuda8.0.27版本，驱动是361。这里，建议你选否，我也选的no。

千万不要改路径，默认就好

安装完成

下一步要把cuda路径添加到环境变量

sudo gedit ~/.bashrc
在最后一个fi后面换行

写入

export PATH=/usr/local/cuda-8.0/binPATH:+:$PATHexportLDLIBRARYPATH=/usr/local/cuda−8.0/lib64{LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存，关闭

source ~/.bashrc

重启

然后应该无坑

输入

nvidia-smi
查看显卡状态

输入

nvcc -V
查看cuda状态

四、安装cudnn

https://developer.nvidia.com/cudnn

N家官网下载，需要注册一个开发者账户，每次下载都要答一个问卷

下载对应的最新版本

解压，cd到新解压出来的cuda文件夹

sudo cp include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp lib64/* /usr/local/cuda-8.0/lib64/

五、安装Keras+Theano

提示，Tensorflow慢成狗，慎用。

先把pip源改成清华镜像：

复制自： 清华大学开源软件镜像站

pypi 镜像使用帮助

pypi 镜像每 5 分钟同步一次。

临时使用

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
注意，simple 不能少, 是 https 而不是 http

设为默认

修改 ~/.pip/pip.conf (没有就创建一个)， 修改 index-url至tuna，例如

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
好，我们成功的避免了pip安装时超时以及30byte/s的情况，下面安装keras就非常简单了

sudo pip install -U –pre pip setuptools wheel
sudo pip install -U –pre numpy scipy matplotlib scikit-learn scikit-image
sudo pip install -U –pre theano
sudo pip install -U –pre keras
theano和keras安装完成，注意，此时只有cpu版本，我们要修改theano配置来实现gpu加速

先把tensorflow改成theano

gedit ~/.keras/keras.json
先import 一次keras

python
import theano
import keras
没报错的话就是成功了

修改theano配置：

gedit ~/.theanorc
写入

[global]
openmp=False
device = gpu
floatX = float32
allow_input_downcast=True
[lib]
cnmem = 0.8
[blas]
ldflags= -lopenblas
[nvcc]
fastmath = True
保存退出

python
import keras
应该会有warning说正在使用cudnn5.1，theano只支持5.，如果有问题就降级cudnn或者升级theano

那么恭喜！你已经成功安装keras和theano，并且开启了gpu支持，可以开始用keras愉快的搞♂深度学习了！

六、安装Tensorflow

完全按照官网来吧：

https://www.tensorflow.org/versions/r0.11/get_started/os_setup.html#pip-installation

Then, select the correct binary to install:

Ubuntu/Linux 64-bit, CPU only, Python 2.7

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

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.0rc2-cp27-none-linux_x86_64.whl

Mac OS X, CPU only, Python 2.7:

$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-0.11.0rc2-py2-none-any.whl

Mac OS X, GPU enabled, Python 2.7:

$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/gpu/tensorflow-0.11.0rc2-py2-none-any.whl

Ubuntu/Linux 64-bit, CPU only, Python 3.4

$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc2-cp34-cp34m-linux_x86_64.whl

Ubuntu/Linux 64-bit, GPU enabled, Python 3.4

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.0rc2-cp34-cp34m-linux_x86_64.whl

Ubuntu/Linux 64-bit, CPU only, Python 3.5

$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc2-cp35-cp35m-linux_x86_64.whl

Ubuntu/Linux 64-bit, GPU enabled, Python 3.5

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.0rc2-cp35-cp35m-linux_x86_64.whl

Mac OS X, CPU only, Python 3.4 or 3.5:

$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-0.11.0rc2-py3-none-any.whl

Mac OS X, GPU enabled, Python 3.4 or 3.5:

$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/gpu/tensorflow-0.11.0rc2-py3-none-any.whl
Install TensorFlow:

Python 2

sudopipinstall−−upgradeTF_BINARY_URL

Python 3

sudopip3install−−upgradeTF_BINARY_URL
So easy！可以在keras里切换后端了。
注意一下keras针对cnn中th和tf卷积核的数据格式

还是在刚才那个修改后端的文件里修改即可。

七、安装MXNet

个人很看好MXNet，简介、速度快、语言多、社区活跃，缺点就是时间段、文档比较少。

先按照官网教程安装

We need git to clone MXNet source code repository to your machine.

Install git if not already installed.

sudo apt-get update
sudo apt-get -y install git

Clone mxnet repository. In terminal, run the commands WITHOUT “sudo”

git clone https://github.com/dmlc/mxnet.git ~/MXNet/mxnet –recursive

Install MXNet dependencies

cd ~/MXNet/mxnet/setup-utils
bash install-mxnet-ubuntu.sh

We have added MXNet Python package path in your ~/.bashrc.

Run below command to refresh environment variables.

$ source ~/.bashrc
下面我们需要修改config.mk文件使其使用gpu

cd MXNet/mxnet

cp make/config.mk .

到mxnet目录编辑config.mk文件

改这几处（文件里本身带教程）

USE_CUDA = 1
USE_CUDA_PATH = /usr/local/cuda
USE_CUDNN = 1
保存
make clean
make -j4
MXNet至此完成

打开python，试验一下

import mxnet as mx
a = mx.nd.ones((2, 3),
… mx.gpu())
print (a * 2).asnumpy()
至此，框架安装结束！

如果我的文档能帮你避掉一两个坑，我将感到无比的欣慰

如有安装过程、代码命令的错误以及侵权等质疑请务必联系我！
以下是安装mxnet0.9.5的命令。
sudo apt-get update
sudo apt-get install -y build-essential git
sudo apt-get install -y libopenblas-dev
sudo apt-get install -y libopencv-dev
git clone https://github.com/dmlc/mxnet.git mxnet –recursive –branch v0.9.5 –depth 1
cd mxnet/setup_utils && bash bash install-mxnet-ubuntu-python.sh