Ubuntu 14 安装 Caffe

来源:互联网 发布:电力科学研究院 知乎 编辑:程序博客网 时间:2024/05/21 18:47

Ubuntu 14 安装 Caffe

因为电脑故障,重装了系统,特此将Caffe安装过程记录下来:

  • 下载Caffe源码
  • 安装Cuda
  • 安装Matlab
  • 使用命令安装众多依赖项
  • 编译Caffe

下载Caffe源码

  • 方式1:用谷歌/baidu 搜“caffe github”,下载zip 包,自行解压
  • 方式2:在终端运行 git clone --recursive https://github.com/BVLC/caffe,直接clone到本地

安装Cuda

  • 下载.deb包(我用的是7.5版本,稍后上传到baidu云盘)
  • 运行以下命令:
    • dpkg -i cuda-repo-ubuntu1404-7-0-local_7.0-28_amd64.deb
    • apt-get update
    • apt-get -y install cuda
  • 更改环境变量
    • 打开终端,输入:
      vim ~/.profile
    • 在文件末尾添加:
      export PATH=/usr/local/cuda-7.5/bin:$PATH
      export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH
      (默认安装基本都是这个路径,以防万一,还是自己顺着这个目录看能否找到)
    • 保存

以上是cuda安装。


安装matlab

  • 下载Matlab(我装的版本是2015a,稍后上传到baidu云盘)
  • 挂载ISO到指定目录:
    • sudo mount matlab2015a.iso /media
  • 切换到指定目录:
    • cd /media
  • 使用脚本自动安装
    • ./install.sh
  • 更改环境变量
    • 打开终端,输入:
      vim ~/.profile
    • 在文件末尾添加:
      export PATH=$PATH:/usr/local/MATLAB/R2015a/bin
      (默认安装基本都是这个路径,以防万一,还是自己顺着这个目录看能否找到)
    • 保存

以上是Matlab安装。


安装python

  • Ubuntu系统自带python
  • 如果想要自己装,推荐使用Anaconda(我是装的Anaconda,装一些自己的库会清爽一些。)
    • 下载Anaconda(稍后传到baidu云盘)
    • 执行以下命令:
      bash Anaconda2-4.1.1-Linux-x86_64.sh
    • 配置环境变量:
      vim .bashrc
      – 文件末尾添加 export PYTHONPATH=/home/archer/caffe-master/python:$PYTHONPATH
      – 保存

以上是python安装。


使用命令安装众多依赖项

  • 先安装基本的(注意到下面已经把opencv一起装了,我这里是2.4.8,用于caffe足够了):

    • 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
  • 安装BLAS:

    • sudo apt-get install libatlas-base-dev
  • 安装其他:
    • sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

以上是依赖项安装。


编译Caffe

  • 在Caffe根目录下,将Makefile.config.example复制,并重命名为Makefile.config

  • 修改Makefile.config

    • 修改matlab路径,比如:
      MATLAB_DIR := /usr/local/MATLAB/R2015a
      (一定要写到bin的上一级目录)

    • 修改cuda路径,比如:

    • 修改python路径
      – 1. 如果使用系统自带python,修改如下:
      PYTHON_INCLUDE := /usr/include/python2.7 \
      /usr/lib/python2.7/dist-packages/numpy/core/include

      PYTHON_LIB := /usr/lib
      – 2. 如果使用Anaconda,修改如下:
      ANACONDA_HOME := /home/archer/anaconda2
      PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
      $(ANACONDA_HOME)/include/python2.7 \
      $(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include \

      PYTHON_LIB := $(ANACONDA_HOME)/lib

  • Caffe根目录下,开始编译
    • make all
      (这里可在命令末尾加上-j8,利用CPU多核编译,能加快编译速度,建议使用;不同机器可能使用的不同,比如有的使用-j16
    • make test
    • make runtest
      (这一步就算有错,似乎也妨碍不大)
    • 提醒一下:如果编译中途出错,下次编译时,记得先执行make clean
  • Caffe根目录下,编译Matlab接口
    • make matcaffe
  • Caffe根目录下,编译python接口
    • make pycaffe

以上就把Caffe基本装好了。

可能自己实际操作中还会遇到一些问题;别急躁,只要以上几步都完成了,那么你已经成功一大半了,遇到问题请耐心在网上搜索(推荐用google),你遇到的问题,别人应该都有讨论到,会得到解决的,加油!

关于Faster Rcnn和FCN框架的安装,可以看这里。

0 0
原创粉丝点击