1. 程式人生 > >凌桑的自我修養--You are the Eternity

凌桑的自我修養--You are the Eternity

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

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安裝成功!!

參考連結