安装环境:Win 10 专业版 64位 + Visual Studio 2015 Community。
记录下自己在有GPU的环境下安装配置MXNet的过程。该过程直接使用MXNet release 的 pre-built 包,没有自己使用CMake编译。网上有很多自己编译的教程,过程都比较繁琐,直接使用release包对新手来说更加简单方便。
选择MXNet的原因是因为看了《Caffe、TensorFlow、MXNet三个开源库的对比》这篇博文,其中指出MXNet相对来说是最易上手的深度学习开源库,所以选择该开源库作为初学工具。
官方文档
Getting Started: http://mxnet.readthedocs.io/en/latest/get_started/
安装
-
配置MXNet的pre-built包
- 下载链接:https://github.com/dmlc/mxnet/releases
- 解压pre-built包到合适的位置,如 E:\MXNet;
- 运行setupenv.cmd,该步骤会帮助添加各种系统环境变量,运行后可以检查下自己的系统环境变量是否正确
- 解压的包里有readme文档,可作为安装过程中的参考。
分为不支持GPU的版本和支持GPU的版本,我的电脑支持GPU,因此选择gpu版本的。
-
配置cuDNN
- 下载链接:https://developer.nvidia.com/cudnn (需要注册为开发者后才能下载)
- 解压cuDNN包,将目录下的bin、include和lib文件夹copy到E:\MXNet\3rdparty\cudnn目录下,覆盖原文件
安装Python2 (version 2.7.11) / Python3 (version 3.5.1)
- 最好python3和python2都安装,虽然发现后面运行默认使用的是python3
- 下载地址:https://www.python.org/downloads/ 选择64位的安装包
- pre-built包是64位的,所以记得安装64位的Python,选择自定义安装,并勾选添加路径到系统变量选项,没有勾选的需要手动添加python安装路径到环境变量。刚开始不小心装了32位的,结果运行报错
安装MXNet的python包
- cmd模式下进入目录 E:\MXNet\python,如下所示:
- 运行python setup.py install,等待安装完毕。安装过程会下载安装python的numpy模块,记得检查是否有错误,如果有错,需要重新运行。
- 我自己在安装的时候因为网络不好,好几次都超时中断了。
- 运行readme文档里的实例代码,看是否正确
参考其他文章,自己编译需要用到以下工具,记录下来备用:
1. OpenCV (version 3.1):OpenCV3.1已经发布,但是只有64位版的。需要32位版本的建议使用OpenCV3.0
下载链接:http://opencv.org/downloads.html
2. OpenBlas
BLAS (Basic Linear Algebra Subprograms):基础线性代数子程序库,是一个应用程序接口(API)标准,说的简单点就是向量、矩阵之间乘加这些运算。
OpenBlas是BLAS的开源优化库,支持CPU多线程。
看到解压出来的 MXNet pre-built包的3rdparty文件夹下有openblas,应该不用自己再下载了。
单独的下载链接为:https://sourceforge.net/projects/openblas/files/latest/download?source=files
相关文档:https://github.com/xianyi/OpenBLAS/wiki/Document
3. CMake工具
非常强大的编译自动配置工具,可以根据不同平台、不同的编译器,生成相应的Makefile或者vcproj项目。
下载链接:https://cmake.org/download/
入门参考:http://www.cnblogs.com/sinojelly/archive/2010/05/22/1741337.html