机器学习 - Ubuntu14.04 x64 搭建 Caffe

前端之家收集整理的这篇文章主要介绍了机器学习 - Ubuntu14.04 x64 搭建 Caffe前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
@H_403_3@
  • @H_403_3@
  • CPU模式
  • 写一下如何在 Ubuntu 14.04上搭建 Caffe

    cpu模式

    这个就是不使用 GPU 加速的,暂时我现在也只使用了这个方式,后面等我买的主机到了,再写 GPU 模式的。

    1. 下载必备依赖
    sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
    sudo apt-get install --no-install-recommends libboost-all-dev

    拷贝到终端里执行,这是一些 Ubuntu 安装 caffe的必备的依赖。

    1. 下载 BLAS
    sudo apt-get install libatlas-base-dev //一个基础线性代数数学库
    1. 下载 Python
    sudo apt-get install python-dev

    一般这个系统自带,不带就安装下。

    1. 继续下载一些依赖
    sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

    继续输入执行。

    1. 下载 caffe
    git clone https://github.com/BVLC/caffe

    然后进入到这个文件加里

    cd caffe/
    cp Makefile.config.example Makefile.config

    然后编辑 Makefile.config这个文件,因为我们没有安装 Cuda所以把第8行的 cpu_ONLY=1的注释取消掉,意思是只使用 cpu.

    1. 编译并测试
    make all
    make test
    make runtest

    如果觉得慢在编译命令后面加-j就是:make all -j这种,如果是多核会加速的。

    如果没报错,就可以使用了。

    然后可以用样例跑一下试试:

    ./data/mnist/get_mnist.sh
    ./examples/mnist/create_mnist.sh
    vi ./examples/mnist/lenet_solver.prototxt

    把里面的solver_mode改为cpu后执行训练脚本:

    谨记一定要在 caffe的根目录下运行,不能cd examples/mnist后执行./create_mnist.sh这样子操作,会失败的!!

    ./examples/mnist/train_lenet.sh

    这样就 OK 了!
    运行图示:

    猜你在找的Ubuntu相关文章