MXNET Windows +Anaconda下安裝教程及Bug解決

前言

MXNET 作為2015年底開始流行的Deep Learning 開源框架， 由於其靈活性、對多種語言的支援、 輕量級的排程引擎已經成為很多DL 研發、學習者的開發框架。 網上也出現了很多的mxnet安裝教程，但隨著mxnet的更新迭代，較早的安裝教程會出現一些問題。 筆者在安裝過程中出現過很多頭疼的Bug, 故針對mxnet 0.9.2（2017年2月更新的mxnet）在windows下的安裝寫一篇教程， 並彙總一些安裝過程中遇到的bug，不定期更新。

1. 安裝環境

1 安裝Cuda
在NVidia

2 安裝Cudnn
Cudnn一定要下載跟cuda對應的版本， 現在cuda 8.0對應的是cudnn 5。注意，有的較早教程提供的是7.5的Cuda，如果安錯了對應的Cudnn以後會出現無法使用Cudnn的Bug
下載地址可以去 Cudnn, 要註冊英偉達賬號
下載後得到Cudnn資料夾（名字也叫cuda）， 分別將裡面的cuda/include、cuda/lib、cuda/bin三個目錄中的內容拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0對應的include、lib、bin目錄下即可
注意是32 or 64位系統

3 Open CV
OpenCV安裝地址為 opencv， opencv 3.0下載下來就好

4 OpenBlas
下載地址為OpenBlas,也是下載下來就好

5 安裝Anaconda
這裡取決於你用的Python編譯器，筆者研發需要安裝了Anaconda 作為IDE。
注意你的python 版本，用Anaconda的話應為2.7版， mxnet暫時是不支援python3的
下載地址Anaconda , 下2.7

6 C++
VS2013及以上都行， 沒有C++11新特性的VS要下載 Visual C++ Compiler Nov 2013 CTP並安裝，然後把 C:\Program Files (x86)\Microsoft Visual C++ Compiler Nov 2013 CTP\ 資料夾內所有拷貝至你的VS2013目錄，如：拷貝至 D:\Microsoft Visual Studio 12.0\VC\ 目錄下覆蓋。

7 記得檢視上述的安裝（一般為1到5步）有沒有新增到系統的path環境變數中， 且檢視版本之間， 如python, anaconda, opencv, cuda之間有沒有衝突

2. 下載MXNET

在完成了 “安裝環境過程後”， 在GitHub 上下載 mxnet， 可以直接down下來，也可以用git fork然後克隆下來。（使用Git請自己百度/Google），下載地址mxnet

下載下來後開啟 mxnet\make\config.mk，查詢並修改為以下引數：
USE_CUDNN = 1
USE_BLAS = openblas

用Cmake和C++編譯MXNET原始碼

其實如果能自己搞到libmxnet.dll是不用cmake 和C++編譯的， 但是自己編譯通過的libmxnet.dll一般沒啥問題，而從別人那copy的就可能發生很多無法描述的bug了，所以筆者還是建議大家儘量自己編譯。

這裡最上面2行填你自己的mxnet安裝路徑
然後點configure， 選你的c++版本， 然後點generator

mxnet_build目錄下會生成mxnet.sln，用vs開啟， 切換成release模式， dubug模式也是可以編譯的，編譯出來也可以用，就是編譯時間很長長長長……..

編譯完成後，在mxnet_build\Release目錄下生成了libmxnet.dll檔案

安裝Python的MXNET庫

將libmxnet.dll檔案copy到 …\mxnet\python\mxnet 路徑下 開啟cmd，切換到當前目錄

輸入python setup.py install 安裝
把 …\Anaconda2\Lib\site-packages\mxnet-0.9.2-py2.7.egg\mxnet 新增到環境變數，中間的.egg是mxnet的版本號

測試MXNET安裝是否成功

開啟anaconda或者python， 執行官網上的程式碼

能跑通證明mxnet安裝成功，且能使用GUP。

Bug彙總

1 執行 python setup.py install 後安裝失敗？

Solution： 如果一直到編譯完libmxnet.dll都沒有出現問題， 而卻無法 setup.py， 請檢查python的版本。 如果已經安了python3， 請把python3路徑下的python.exe改個名字， 如果用anaconda也同樣。 最後請檢查python或anaconda的環境變數， mxnet是無法用python3 or anaconda3 的。 在執行時用 管理員許可權執行。

2 在跑 Import mxnet as mx , a = mx.nd.ones((2,3), mx.gpu（）)程式碼時， 最後print 時全為0？

Solution： 沒有用上gup，檢查gup是否符合cuda， 把程式碼改成a = mx.nd.ones((2,3), mx.cpu（）)試試用cup有沒有問題。如果CPU可以正常執行，請仔細排查安裝教程1-7涉及的問題，或者轉到問題3

3 “ImportError: cannot import name libinfo” python import error on windows？

Solution： 這種情況一般為libinfo檔案沒有找到dll， 請把安裝的open, openblas， cudnn 等的.dll檔案copy到 你的python/anaconda 的mxnet庫下， 例如：筆者用anaconda2則copy到 E:\Anaconda2\Lib\site-packages\mxnet-0.9.2-py2.7.egg\mxnet 下
這裡注意32 or 64位

4 Import Error: No module named cv2?

Solution： 在跑mxnet自帶的pre_trained model例子時出現的問題， 原因是缺少了opencv 的cv2檔案，將OpenCV安裝目錄裡的Python資料夾內的cv2.pyd複製到Python安裝目錄裡Lib中site-packages內即可解決。

5 Failed loading Op conv_conv1 of type Convolution: Cannot find argument ‘cudnn_tune’ ？

Solution：也是在跑pre_trained model時出現的bug, 原因是cudnn沒有用上， 請按教程全部重新安裝。（記得刪除之前的無效版， 否則會起衝突）