网站首页Python相关

centos7安装caffe(仅cpu)

发布时间:2017-02-06编辑:fc不将就阅读(1412

    首先需要安装OPENCV:传送门

    安装依赖

    安装elep源及部分依赖包
    $yum install epel-release -y
    $yum install atlas-devel protobuf-devel leveldb-devel snappy-devel 
    $yum install opencv-devel boost-devel hdf5-devel gflags-devel glog-devel lmdb-devel
    安装OpenBLAS

    下载地址:https://sourceforge.net/projects/openblas/files/v0.2.15/

    默认安装在:/opt/OpenBLAS

    $cd /var/local
    $tar -xzvf OpenBLAS-0.2.19.tar.gz
    $make
    $make install

    配置环境变量:

    $vi /etc/profile
    #添加如下内容
    export LD_LIBRARY_PATH=/opt/OpenBLAS/lib/:$LD_LIBRARY_PATH
    
    $source /etc/profile

    安装caffe

    下载源码包
    $cd /var/local
    $git clone https://github.com/BVLC/caffe.git
    安装python依赖
    $cd /var/local/caffe/python
    $pip install -r requirements.txt
    编译caffe
    $cd /var/local/caffe
    $cp Makefile.config.example Makefile.config
    $vim Makefile.config
    #修改如下内容,根据实际情况修改
    CPU_ONLY := 1
    BLAS := open
    BLAS_INCLUDE := /opt/OpenBLAS/lib
    BLAS_LIB := /opt/OpenBLAS/lib
    
    $make all
    #对编译结果进行测试
    $make test
    #编译python caffe
    $make pycaffe
    #验证caffe可用
    $cd python
    $python
    >>> import caffe
    >>> caffe.__version__
    '1.0.0-rc3'

    添加python caffe路径环境变量:

    $vim /etc/profile
    #添加如下内容
    #CAFFE SETTING
    export CAFFE_HOME=/var/local/caffe/
    export PYTHONPATH=$PYTHONPATH:$CAFFE_HOME/python
    
    $source /etc/profile