首先下载mxnet安装包:
git clone --recursive https://github.com/dmlc/mxnet
进入mxnet目录,并将mxnet/make/config.mk复制到mxnet/下:
cd mxnet
cp make/config.mk .
修改config.mk文件的行为:
USE_CUDA = 1
USE_CUDA_PATH = USE_CUDA_PATH = /usr/local/cuda-8.0
USE_CUDNN = 1
编译:
make -j8
安装:
sudo python setup.py install
安装之后运行SVM手写字体分类:
python example/svm_mnis/svm_mnist.py