您的位置:首页 > 其它

windows 10下安装GPU版MXNet 0.7.0

2016-12-05 21:52 323 查看

前置条件

已经安装:

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

打开cmake 把mxnet中的CmakeList.txt 拖入,

确保USE_CUDA, USE_CUDNN,
USE_MXNET_LIBNAMING, USE_OPENCV
, USE_OPENMP,全选上

改生成路径“where to bulid the binaries ”为mxnet/build

点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

打开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。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐