凌桑的自我修養--You are the Eternity
阿新 • • 發佈:2018-12-30
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安裝成功!!