[TOC]
简介
caffe是一个清晰,可读性高,快速的深度学习框架。作者是贾扬清,加州大学伯克利的ph.D,现就职于Facebook。caffe的官网是http://caffe.berkeleyvision.org/。
安装
环境
- Ubuntu 16.04 LTS
- CUDA8.0
参考资料
- 官方安装指南:Ubuntu Installation
注: 对于 Ubuntu >= 17.04 的Ubuntu系统,安装将变得极为简单,通过Ubuntu的包管理器安装即可,这里不作说明。
准备
基础环境配置
- 系统安装,可移步本人博客:U盘系统盘制作与系统安装(详细图解)
- CUDA安装,可移步本人博客:Ubuntu 16.04 LTS + CUDA8.0 + cudnn6.0 或 NVIDIA DIGITS 学习笔记(NVIDIA DIGITS-2.0 + Ubuntu 14.04 + CUDA 7.0 + cuDNN 7.0 + Caffe 0.13.0)
依赖库安装
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
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install libatlas-base-dev
构建与安装
获取源码
从 这里 (或终端输入命令 git clone https://github.com/BVLC/caffe.git
)下载caffe源码并解压,修改文件夹名为 caffe
。
For Python
如果你需要使用Caffe 的 Python 接口,你还需要安装Python的一些库,进入caffe根目录下的python目录,执行:
for req in $(cat requirements.txt); do sudo pip install $req; done
然后添加 PYTHONPATH
环境变量:终端输入 sudo gedit ~/.bashrc
打开 .bashrc
文件,文件末尾加入(注意修改为你的caffe绝对路径):
#caffe python
export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH
如果你使用 Anaconda ,下面修改 Makefile.config
文件时,注意自行 启用 Anaconda,约73行。
修改配置文件
这一步修改 Makefile.config
文件,以满足自己的需求,如:
#启用cuDNN加速
USE_CUDNN := 1
另外,在Ubuntu16.04上安装caffe还需要修改以下几处,否则会报出各种错误,具体参见问题解决:
将如下几处代码:
#第一处:CUDA PATH
CUDA_DIR := /usr/local/cuda
#第二处:PYTHON PATH
# Whatever else you find you need goes here.
#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
改为:
#第一处:CUDA PATH,改为:
CUDA_DIR := /usr/local/cuda-8.0
#第二处:PYTHON PATH,添加hdf5库,改为:
# Whatever else you find you need goes here.
#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/
构建安装
进入caffe的根目录,终端执行如下命令,不报错误,代表安装成功!( -j$(nproc)
代表使用最大的线程编译,当然也可以手动指定,如 make -j4
; make py
是用来编译caffe的python接口,如不需要可忽略。)
make all -j$(nproc)
make pycaffe -j$(nproc)
make test -j$(nproc)
make runtest -j$(nproc)
添加环境变量
终端输入 sudo gedit ~/.bashrc
打开 ".bashrc" 文件,在文件末尾加入如下代码并保存:
# Set Caffe environment
export CAFFE_ROOT=/home/liu/sfw/dlapp/caffe/
#caffe python
export PYTHONPATH=/home/liu/sfw/dlapp/caffe/python/:$PYTHONPATH
然后重新打开一个终端,或者输入 source ~/.bashrc
加载新的环境变量。
验证Python接口
为验证安装成功,终端输入 ipython
或 python
进入Python解释器环境,输入: import caffe
,不报错误代表caffe的Python接口可以使用。
问题解决
安装
hdf5
- 现象
In file included from src/caffe/solvers/sgd_solver.cpp:5:0:
./include/caffe/util/hdf5.hpp:6:18: fatal error: hdf5.h: No such file or directory
compilation terminated.
Makefile:581: recipe for target '.build_release/src/caffe/solvers/sgd_solver.o' failed
- 解决
首先确保前面安装依赖库时安装了 libhdf5-serial-dev
,然后修改 Makefile.config
文件(约95行):
# Whatever else you find you need goes here.
#INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/lib/x86_64-linux-gnu/hdf5/serial/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial/
No module named caffe
请确认是否执行 make pycaffe
且没有报错,如果是请再检查环境变量的设置。