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

官网下载(http://sourceforge.net/projects/opencvlibrary/files/opencv-win/3.0.0/opencv-3.0.0.exe/download)。

(2) OpenBLAS

官网下载(https://sourceforge.net/projects/openblas)。

(3) CuDNN

官网下载(https://developer.nvidia.com/cudnn)。

(4) CUDA

官网下载(https://developer.nvidia.com/cuda-downloads)并安装。

3. CMake编译

下载并安装最新版的CMake,官网下载(https://cmake.org/download)。然后如图填写参数&打勾。
Windows下mxnet的编译与安装_第1张图片

点击Configure进行配置,generator选项中VS版本随意,但一定要选Win64。
如果报错找不到OpenBLAS,就需要我们手动添加OpenBLAS的路径:
Windows下mxnet的编译与安装_第2张图片

再点Configure,如果报错找不到OpenCVConfig.cmake,再次添加OpenCV的路径,保证该路径下有OpenCVConfig.cmake:
Windows下mxnet的编译与安装_第3张图片

再次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的安装工作就完成了。

你可能感兴趣的:(mxnet)