Windows下mxnet的编译与安装

    xiaoxiao2021-12-14  19

    本文将给出在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)。然后如图填写参数&打勾。

    点击Configure进行配置,generator选项中VS版本随意,但一定要选Win64。 如果报错找不到OpenBLAS,就需要我们手动添加OpenBLAS的路径:

    再点Configure,如果报错找不到OpenCVConfig.cmake,再次添加OpenCV的路径,保证该路径下有OpenCVConfig.cmake:

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

    转载请注明原文地址: https://ju.6miu.com/read-963002.html

    最新回复(0)