ubuntu16.04安装mxnet

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u011961856/article/details/77983663

首先下载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
这里写图片描述

阅读更多
博主设置当前文章不允许评论。
换一批

没有更多推荐了,返回首页