windows 10下安装GPU版MXNet 0.7.0

前置条件

已经安装:

• visual studio 2013(参照mxnet官网，安装“ Visual C++ Compiler Nov 2013 CTP.”使其支持C++11,如果装不上需要卸载一些东西,印象中是“Microsoft 2013 VC++ 2013 Redistrubutable x86、x64”)

• cuda 8.0

• git（最新版）
• anaconda2（最新版）
• cmake（最新版）

下载包列表

• opencv3.0 （3.1亦可）
• openblas(笔者用0.2.19)
• openblas 的依赖mingw64_dll.zip (这个很重要，卡了很久，不拷贝到mxnet的安装目录会报“WindowsError: [Error 126]”错误。用anaconda 安装的mingw 貌似无效)https://sourceforge.net/projects/openblas/files/v0.2.15/mingw64_dll.zip/download
• cudnn5.1(5.0亦可)

安装

1. 用git下载mxnet ，注意 –recursive

git clone https://github.com/dmlc/mxnet.git  --recursive

2、编译mxnet

1. 打开cmake 把mxnet中的CmakeList.txt 拖入，
2. 确保USE_CUDA, USE_CUDNN, USE_MXNET_LIBNAMING, USE_OPENCV, USE_OPENMP，全选上
3. 改生成路径“where to bulid the binaries ”为mxnet/build
4. 点Configure 按钮，错的地方填下面的东西，再点Configure 按钮，直到没有红色为止，再点Generate按钮
   
   • OpenCV_DIR ： opencv-3.1.0/build
   • OpenCV_CONFIG_PATH ：OpenCVConfig.cmake文件
   • CUDNN_INCLUDE： cuda/include
   • CUDNN_LIBRARY ： cuda/lib/x64/cudnn.lib
   • OpenBLAS_INCLUDE_DIR should be ：OpenBLAS-v0.2.19/include
   • OpenBLAS_LIB ：OpenBLAS-v0.2.19/lib/libopenblas.dll.a
5. 打开build文件夹下的的mxnet,sln文件，改变生成项目为release版，右击“ALL_BUILD”项目生成，大约10几分钟生成完毕

3、安装python版mxnet

cd python
 python setup.py install

4、拷贝相应的dll到mxnet的安装目录

打开\Anaconda2\Lib\site-packages\mxnet-0.7.0-py2.7.egg\mxnet
拷贝如下的dll到该目录

5、把 “C:\Anaconda2\Lib\site-packages\mxnet-0.7.0-py2.7.egg\mxnet”添加到环境变量

6、跑手写库如下

总结

卡了很久在WindowsError: [Error 126]”错误上，网上都没说到点子上。主要因为缺少 mingw64_dll.zip。