Ubuntu 16.04TLS 安裝MXNet過程(含OpenCV 3.1.0配置)

雖說MXNet安裝過程相對簡單,但是實際在操作的時候也遇到了不少坑,在這里簡單記錄一下。

我用的顯卡型號是GeForce GTX 1060。顯卡的驅動安裝和CUDA以及CUDNN的配置因為之前已經搞定了,這里就不贅述(整個過程也是踩遍了各種坑坑洼洼,有空再另寫一篇)。CUDA版本查詢方式如下圖所示,CUDNN版本為V5.0.5。

現在可以開始入坑了

1安裝OpenCV

參考:手把手教你,在Ubuntu上安裝OpenCV 3.0 和 Python 2.7+

電腦里的python有兩個版本,2.7和3.5,這里安裝的OpenCV的版本的3.1.0,注意對應的opencv_contrib也要裝3.1.0版本。

打開終端窗口,更新apt-get包管理器,升級所有預安裝包:

$ sudo apt-get update
$ sudo apt-get upgrade

安裝我們的開發工具和包:

$ sudo apt-get install build-essential cmake git pkg-config

OpenCV 需要從磁盤中加載不同格式的圖片,如JPEG,TIFF, PNG等等.所以我們需要安裝我們的圖像I/O工具包:

$ sudo apt-get install libjpeg8-dev libtiff4-dev libjasper-dev libpng12-dev

另外,OpenCV是用GTK開發包來顯示GUI, 即用戶圖形界面,所以我們要安裝這個開發包:

$ sudo apt-get install libgtk2.0-dev

OpenCV還必須處理視頻流和單個幀,下面就是我們需要的安裝包:

$ sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev libv4l-dev

OpenCV還包含一些內部優化工具:

$ sudo apt-get install libatlas-base-dev gfortran

安裝python包管理器pip:

$ wget https://bootstrap.pypa.io/get-pip.py
$ sudo python get-pip.py

安裝virtualenv和virtualenvwrapper. 用來分割python虛擬環境. 這不是必須的, 很多教程強烈推薦,但我嫌麻煩暫時沒裝,也沒問題:

$ sudo pip install virtualenv virtualenvwrapper
$ sudo rm -rf ~/.cache/pip

現在我們有了virtualenv和virtualenvwrapper, 我們要更新我們的~/.bashrc? 文件:

# virtualenv and virtualenvwrapper
$ export WORKON_HOME=$HOME/.virtualenvs
$ source /usr/local/bin/virtualenvwrapper.sh

為了使~/.bashrc 文件生效 , 你可以用以下這些方法的其中之一(1) 注銷后重新登錄, (2)關閉終端開一個新終端, (3)直接使得~/.bashrc文件在當前生效:

$ source ~/.bashrc

最后我們生成名字叫cv的虛擬開發環境:

$ mkvirtualenv cv

安裝python開發工具(一般都會裝好python,這里以2.7版本為例,事實上我的電腦有兩個版本2.7和3.5):

$ sudo apt-get install python2.7-dev

安裝numpy:

$ pip install numpy

預備環境終于都搞定啦, 我們進入正題, 按照OpenCV 3.1.0:

$ cd ~
$ git clone https://github.com/Itseez/opencv.git
$ cd opencv
$ git checkout 3.1.0

有一些牛叉的算法如SIFT, SURF, 等等 在opencv_contrib里面, 所以我們要安裝它來支持OpenCV:

$ cd ~
$ git clone https://github.com/Itseez/opencv_contrib.git
$ cd opencv_contrib
$ git checkout 3.1.0

注意: opencv_contrib和OpenCV版本要一致

是時候build OpenCV(坑來了):

$ cd ~/opencv
$ mkdir build
$ cd build
$ cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local -D WITH_TBB=ON -D WITH_V4L=ON -D WITH_QT=ON D BUILD_EXAMPLES=OFF -D WITH_OPENGL=ON -D? WITH_CUDA=OFF -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules ..?

這里有幾個關鍵的地方要注意,WITH_CUDA=OFF(不太明白為什么要把CUDA編譯關了,但是試過如果不關,我這里opencv編譯會出現error...可以先不關試試,如果出現問題再關掉重新編譯。BUILD_EXAMPLES=OFF,這個是要關掉的,不關好像3.1.0的版本編譯不通過)

最后, 編譯! :

$ make -j8

編譯沒問題的話, 就可以安裝了:

$ sudo make install
$ sudo ldconfig

如果安裝無誤, OpenCV現在已經安裝在/usr/local/lib/python2.7/site–packages中了. 但是考慮到我們的虛擬環境cv還沒有OpenCV, 我們需要建立一個軟鏈(我沒做這步):

$ cd ~/.virtualenvs/cv/lib/python2.7/site-packages/
$ ln -s /usr/local/lib/python2.7/site-packages/cv2.so cv2.so

恭喜 ! 你完成了在Ubuntu上安裝OpenCV 3.1:

所有剩下的就是驗證一下啦:

$ workon cv
$ python
>>> import cv2
>>> cv2.__version__
'3.1.0'

OK,可以進入下一階段了

2安裝MXNet

MXNet 的安裝基本上是官方的標準流程.

$ sudo apt-get install -y build-essential git libatlas-base-dev
$ git clone --recursive https://github.com/dmlc/mxnet
$ cd mxnet

在編譯之前, 可以修改 MXNet 的 config.mk 文件, 來滿足個性化的需求.

$ cd mxnet
$ cp make/config.mk .
$ vi -nw config.mk # 或者使用其它順手的編輯器

我改了四個地方:

USE_CUDA = 1
USE_CUDA_PATH = /usr/local/cuda
USE_CUDNN = 1
USE_BLAS = atlas

保存后就可以編譯了

$ make -j8

之前因為沒有裝opencv,所以編譯一直出錯。opencv裝完后,編譯到最后出現一個問題:

g++ example.cpp -o example `pkg-config --libs --cflags opencv`
/usr/bin/ld: 找不到 -lippicv
collect2: error: ld returned 1 exit status

大概是這個錯, /usr/bin/ld: 找不到 -lippicv(ubuntu 16.04 LTS 安裝opencv-3.1.0,含完整安裝步驟)?這里有三個解決方案,我用第三個方案解決了這個問題,感謝作者。

從opencv.pc中可以看出,pkg-config加載庫的路徑是/usr/local/lib,我們去這這個路徑下看看,發現沒有-lippicv對應的庫,別的選項都有對應的庫,然后我們把-lippicv對應的庫(libippicv.a)放到這個路徑下就好啦了。

我的liboppicv.a在~/opencv/3rdparty/ippicv/unpack/ippicv_lnx/lib/intel64這個路徑下。你的也在你自己opencv文件夾的對應路徑下。

先cd 到上面這個路徑下,然后cp liboppicv.a /usr/local/lib 將這個庫文件復制到/usr/local/lib下就好了。

再試試,發現“/usr/bin/ld: 找不到 -lippicv

collect2: error: ld returned 1 exit status”這個錯誤已經解決。

為了加速 GPU 計算, 我們忽略顯卡的兼容, 只支持當前的顯卡(其他教程建議的,以后重新編譯可以試試)

首先要查詢當前的顯卡的 compute capacity, 然后修改 mshadow 的編譯選項, 例如, ?1080 的編譯選項如下:

# mxnet/mshadow/make/mshadow.mk
MSHADOW_NVCCFLAGS = -gencode arch=compute_61,code=sm_61

編譯成功后,安裝python支持:

$ cd ~/mxnet/python
$ python setup.py install

可以運行它的樣例程序試試效果:

cd mxnet/example/image-classification
python train_mnist.py

如果要實現GPU加速

python train_mnist.py --gpus 0 --network lenet

好了,簡要介紹完了~

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,546評論 6 533
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,570評論 3 418
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,505評論 0 376
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,017評論 1 313
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,786評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,219評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,287評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,438評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,971評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,796評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,995評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,540評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,230評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,662評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,918評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,697評論 3 392
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,991評論 2 374

推薦閱讀更多精彩內容