MXNET Windows Compilation installation (Python)
This article only records mxnet installation under Windows, more environment configuration please visit Official document: http://mxnet.readthedocs.io/en/latest/how_to/build.html
Compile target:
Necessary:
- Support c++11,g++>=4.8
- Blas libraries, such as Libblas, Libblas, Openblas Intel MKL
Optional conditions:
- CUDA Toolkit >= v7.0 to run on Nvidia GPUs
- Requires GPU with support for Compute Capability >= 2.0
- CUDNN to accelerate the GPU computation (only CUDNN 3 is supported)
- OpenCV for Image Augmentation
Steps
First, strengthen the VS2013 to enable it to support the c++11 feature.
- Download Install: Visual C + + Compiler-Nov CTP.
- Copy the files under the installation directory to the VS2013 appropriate installation directory, for example: Copy C:\Program files (x86) \microsoft Visual C + + Compiler All files to C:\Program file S (x86) \microsoft Visual Studio 12.0\VC It is best to back up the original files.
Second, install a third-party library.
Includes OpenCV, CuDNN, and Openblas (this is ignored if MKL is already installed).
Finally, using CMake to create a vs project, CMake needs to be pre-installed.
Note that you should choose whether or not to Win64 according to your machine, otherwise you will not find Cublas when you configure OpenCV.
After clicking Configure, you need to configure the OpenCV, Openblas, and CUDNN paths and follow the prompts next.
When prompted configure done, click Generate to build the solution.
Tip generate done, open the solution in VS to compile.
If the hint opencv2/opencv.hpp cannot be found, then in VC + + of the project properties, add the file directory in the Include directory (found in the OpenCV installation directory), and similar issues are resolved.
Python Package Installation
Need python>=2.7 and NumPy. Install numpy using the PIP command on the Windows command line:
Pip Install NumPy
Verify that the installation is successful:
Python example/image-classification/train_mnist.py
Install (place the desired library file in the specified location):
Python setup.py Install
or set the environment variable PYTHONPATH to/<rootpathtoproject>/mxnet/python
Train MLP on MNIST
Now train a MLP to get a quick look at the process of training a network and the associated Python interface.
Importmxnet as MX#Step 1 Configuration Training setTrain =Mx.io.MNISTIter (Image="Mnist/train-images-idx3-ubyte", the label="Mnist/train-labels-idx1-ubyte", Batch_size= 128, Data_shape= (784, ))#Step 2 Configuring the validation setval =Mx.io.MNISTIter (Image="Mnist/t10k-images-idx3-ubyte", the label="Mnist/t10k-labels-idx1-ubyte", Batch_size= 128, Data_shape= (784, ))#Step 3 Configure the network, here a simple three layerdata = Mx.symbol.Variable ('Data') FC1= mx.symbol.FullyConnected (data = data, num_hidden=128) Act1= mx.symbol.Activation (data = FC1, act_type="Relu") FC2= mx.symbol.FullyConnected (data = act1, Num_hidden = 64) Act2= mx.symbol.Activation (data = FC2, act_type="Relu") FC3= mx.symbol.FullyConnected (data = Act2, num_hidden=10) MLP= mx.symbol.SoftmaxOutput (data = FC3, name ='Softmax')#Step 4 feedforward Network configurationModel =Mx.model.FeedForward (Symbol=MLP, Num_epoch= 20, Learning_rate=. 1)#Step 5 FittingModel.fit (X = train, Eval_data =val)#after step 6 training is completed, use the trained model to predictModel.predict (X = val)
Pits ~ ~
Mxnet Windows configuration