深度学习框架MXnet安装

2017-04-29 20:50:262157 views

1.基本依赖包安装

    sudo apt-get update     
    sudo apt-get install -y build-essential git libblas-dev libopencv-dev
    sudo apt-get install libatlas-base-dev

2.CUDA和CUDNN安装

教程分别如下

http://zhuhd.win/TP/index.php?s=/Home/Index/article/id/39

3.编译支持GPU的MXnet

    git clone --recursive  https://github.com/dmlc/mxnet    
    cd mxnet
    cp make/config.mk config.mk
    sudo gedit config.mk

USE_CUDA = 1

USE_CUDA_PATH = /usr/local/cuda

实测上面这个路径不要加版本号，否则会找不到cuda.h

sudo make -j4

4.安装Python支持

    cd python    
    sudo apt-get install python-numpy python-scipy python-matplotlib
    sudo python setup.py install

5.测试Mnist手写体识别实例

    cd mxnet/example/image-classification
    python train_mnist.py 
    python train_mnist.py --gpus 0 --network lenet