MXNet有提供Windows下编译好的文件,不用自己编译。
依赖
- Python,2.7或3.5,建议使用Anaconda安装Python,自带了很多有用的库
- VS2015,或VC2015分发包,对应VC14版本
- CUDA,可选,若支持GPU需安装
- CUDNN,可选,若支持GPU建议安装
使用Prebuilt的文件安装
1. 从Github下载编译好的文件,下载地址,注意该Windows Nightly Build不在MXNet官方Repo下。
依次下载上图所示的两个文件,包2根据是否支持gpu自行选择。将下载下的两个包解压,用包2解压之后的文件覆盖包1的文件,本示例的路径为C:\MXNet
2. 在解压后文件夹中有一个名为setupenv.cmd的文件,运行该文件,会自动配置所需的环境变量。
3. 安装MXNet的Python包,进入C:\MXNet\python,在命令行运行命令python setup.py install来安装包
4. 测试MXNet是否安装成功,在Python下导入mxnet:
`import mxnet as mx`