美文网首页
Ubuntu 16.04 下安装Caffe环境

Ubuntu 16.04 下安装Caffe环境

作者: 雲月冠上 | 来源:发表于2017-07-12 23:09 被阅读0次

    安装依赖

    sudo apt install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
    sudo apt install --no-install-recommends libboost-all-dev
    sudo apt install libgflags-dev libgoogle-glog-dev liblmdb-dev libatlas-base-dev
    sudo apt install libgflags-dev libblas-dev liblmdb-dev libgflags-dev

    编译

    一、去载源码

    github 下载

    二、配置

    cp Makefile.config.example Makefile.config

    如果是CPU模式,则配置为

    CPU_ONLY := 1

    解决hdf5错误问题

    INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
    LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial

    三、编译
    依次执行命令

    make all -j8
    make test -j8
    make runtest -j8

    注意:-j8是指使用8核进行编译,根据配置调整

    编译python接口

    一、安装依赖

    sudo apt install python-pip
    sudo apt install python-numpy
    sudo apt install gfortran
    for req in $(cat python/requirements.txt); do pip install $req; done
    sudo pip install -r python/requirements.txt

    二、编译

    make pycaffe

    三、配置环境变量

    export PYTHONPATH=/home/ubuntu/caffe/python:$PYTHONPATH

    四、测试

    运行python
    >> import caffe
    如果没有报错,说明caffe安装全部完成!

    相关文章

      网友评论

          本文标题:Ubuntu 16.04 下安装Caffe环境

          本文链接:https://www.haomeiwen.com/subject/imeihxtx.html