FunnyWeb

据说这里有很多神奇的东西...

深度学习框架MXnet安装

2017-04-29 20:50:26891 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/38

    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

image.png