Ubuntu14.04下MXNet安装

版权声明：本文为博主原创文章，转载请注明出处：http://blog.csdn.net/eternity1118_。 https://blog.csdn.net/eternity1118_/article/details/55192921

Ubuntu14.04下MXNet安装

MXNet是一个轻量化分布式且可移植的深度学习框架，当前支持Python、R、Julia以及Scala等语言，但由于目前使用Python者众多，所以本文只介绍for Python的相关安装，其它语言的安装可以参考官网。

安装流程

step0 安装基本的依赖包

打开终端，键入命令：

sudo apt-get update
sudo apt-get install –y build-essential git libatlas-base-dev libopencv-dev

step1 安装CUDA

如果你没有NVDIA的显卡，请直接跳过step1和step2；
a. 下载对应版本的cuda：
https://developer.nvidia.com/cuda-toolkit
b. 安装：
键入命令：

step2 安装cudnn

a. 下载cudnn：
https://developer.nvidia.com/cudnn
b. 安装：
解压，并将include的头文件和lib64的所有lib文件都拷贝到cuda的安装路径下对应的文件夹中；
键入命令：

tar xvzf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp –p cuda/include/cudnn.h /usr/local/cuda/include
sudo cp –p cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
sudo ldconfig

step3 下载MXNet源码

直接从github上克隆下源码；

git clone https://github.com/zhleternity/mxnet --recurive

step4 安装OpenCV

a. 安装opencv的依赖项：

sudo apt-get install libgtk2.0-dev libavodec-dev libavformat-dev libjpeg62-dev cmake libswscale-dev libjasper-dev

b. 下载opencv并解压：
http://opencv.org/
c. 安装opencv，在opencv根目录下：

mkdir build
cd build
cmake ..
make
sudo make install

d. 配置设置：

sudo gedit /etc/ld.so.conf.d/opencv.conf

在打开的文件中添加opencv的链接库路径：

/usr/local/lib

e. 继续执行：

sudo ldconfig
sudo gedit /etc/bash.bashrc

在打开的文件中添加如下语句：

PKG_CONFIG_PATH=$PKG_CONFIG_PATH:/usr/local/lib/pkgconfig
export PKG_CONFIG_PATH

step5 安装MXNet for Python

编译MXNet源码：

cd mxnet
sudo cp make/config.mk .
sudo gedit config.mk

这里要注意：如果你使用cuda加速，就在打开的配置文件中更改一些设置，如：

如果不使用，则无需更改；
编译：

make

或者

make –j8

b．安装MXNet的Python支持：
1）安装必要的工具包

sudo apt-get install –y python-setuptools python-pip python-numpy python-scipy python-matplotlib

2）安装

cd python
sudo python setup.py install develop –-user

3）配置

sudo gedit ~/.bashrc

在打开的文件中添加MXNet下的python路径：

export PYTHONPATH=~/mxnet/python

（注意，这个路径需要根据你的mxnet的存放目录来设置；）

step6 安装Python的相关库

sudo pip install jupyter
sudo pip install graphviz

根据工作需要，后续可以安装其它库；

step7 测试MXNet是否安装成功

新建一个终端，进入Python，键入：

python
>>>import mxnet

这条命令如果未出现任何问题，则说明mxnet安装成功！！

参考链接

MXNet官网：
http://mxnet.io/get_started/ubuntu_setup.html
csdn博客：
http://blog.csdn.net/u012759136/article/details/50196685