1. 程式人生 > >Windows下mxnet的編譯與安裝

Windows下mxnet的編譯與安裝

本文將給出在windows下mxnet的安裝方法:

執行環境:
Windows 8.1-64 bit
Python 3.5.2-64 bit
Visual Studio 2013 Ultimate

1. 獲取mxnet

開啟本地Github Shell,執行git clone --recursive https://github.com/dmlc/mxnet,然後將程式碼存到本地路徑R:\MXNet下。

2. 安裝幾個第三方依賴項

需要OpenCV3, OpenBLAS, CuDNN, CUDA. 前三項我已經配置好並上傳到網盤連結(密碼:5qmg),可以直接下載放到R:\MXNet\mxnet_thirdparty路徑下。

(1) OpenCV3

(2) OpenBLAS

(3) CuDNN

(4) CUDA

3. CMake編譯

點選Configure進行配置,generator選項中VS版本隨意,但一定要選Win64。
如果報錯找不到OpenBLAS,就需要我們手動新增OpenBLAS的路徑:
OpenBLAS

再點Configure,如果報錯找不到OpenCVConfig.cmake,再次新增OpenCV的路徑,保證該路徑下有OpenCVConfig.cmake:
OpenCV

再次Configure,應該就會Configuring done了。然後Generate就可以生成可執行的sln檔案,CMake完成。

4. 生成libmxnet.dll

用VS開啟MXNet\mxnet_build路徑下的mxnet.sln進行Release編譯,完成之後就會在MXNet\mxnet_build\Release中生成libmxnet.dll檔案。這個步驟耗時較長,而且進度條並非線性變化,所以只要沒報錯,大家耐心等待就好。

5. 安裝python的mxnet庫

把第4步生成的libmxnet.dll檔案拷貝至R:\MXNet\mxnet\python\mxnet目錄內。執行cmd,切換至R:\MXNet\mxnet\python目錄下,執行python setup.py buildpython setup.py install

。在執行這兩句命令時會檢測當前python中有沒有安裝numpy庫,如果沒有會自動easy_install。所以要麼確保python中有setuptools庫,要麼先用easy_install或pip安裝numpy。

6. Last Step

這時已經成功安裝了mxnet庫,但是在import mxnet時會報錯,找不到libopenblas.dll。我們將R:\MXNet\mxnet_thirdparty\OpenBLAS\bin目錄下的4個ddl檔案複製到python.exe的目錄下即可解決這個問題。

至此,mxnet的安裝工作就完成了。