Ubuntu16.04中Caffe的安装

1. 首先安装一些依赖库

$ sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
$ sudo apt-get install -y libatlas-base-dev
$ sudo apt-get install -y --no-install-recommends libboost-all-dev
$ sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev
$ sudo apt-get install -y python-pip python-dev python-numpy python-scipy
$ sudo apt-get install -y libopencv-dev

2.下载caffe.git 到download目录

$ git clone https://github.com/BVLC/caffe.git
$ cd ~/download
$ cp -r ./caffe ~/

3.进入 caffe 中复制配置文件并重命名

$ cd ~/caffe
$ sudo cp Makefile.config.example Makefile.config

4.修改配置文件

$ sudo vim Makefile.config

根据个人需要进行修改

a.若使用 cudnn,则将#USE_CUDNN := 1
修改成:USE_CUDNN := 1
注意:GPU运算能力(GPU Compute Capability )3.0以上才支持CUDNN

b.若使用的 opencv 版本是 3 的,则将#OPENCV_VERSION := 3
修改为:OPENCV_VERSION := 3
c.若要使用 python 来编写 layer,则将#WITH_PYTHON_LAYER := 1
修改为:WITH_PYTHON_LAYER := 1
d.重要的一项 :将# Whatever else you find you need goes here.下面的
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
修改为:
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 /usr/lib/x86_64-linux-gnu/hdf5/serial
e.还要更改
USE_OPENCV: = 1
USE_LEVELDB: = 1
USE_LMDB: = 1

5. 修改 makefile 文件

打开 Makefile 文件,在 415 行,将:
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
替换为:
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

6.修改 host_config.h 文件

sudo vim /usr/local/cuda/include/host_config.h

将其中的第 119 行注释掉:

#error-- unsupported GNU version! gcc versions later than 5 are not supported!

改为:

//#error-- unsupported GNU version! gcc versions later than 5 are not supported!

7.执行编译和测试命令

在~/ caffe/目录下

$ make all -j4
$ make test -j4
$ make runtest -j4

问题:
Makefile:532: recipe for target 'runtest' failed
make: *** [runtest] 已放弃 (core dumped)

这个问题就是,电脑不支持cudnn,再前面的Makefile.config里面去掉CUDNN就可以了。

另外,
编译过程中出现警告:
nvcc warning : The 'compute_20', 'sm_20', and 'sm_21' architectures are deprecated, and may be removed in a future release (Use -Wno-deprecated-gpu-targets to suppress warning).
解决:修改Makefile.config文件

CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
                -gencode arch=compute_20,code=sm_21 \
                -gencode arch=compute_30,code=sm_30 \
                -gencode arch=compute_35,code=sm_35 \
                -gencode arch=compute_50,code=sm_50 \
                -gencode arch=compute_52,code=sm_52 \
                -gencode arch=compute_60,code=sm_60 \
                -gencode arch=compute_61,code=sm_61 \
                -gencode arch=compute_61,code=compute_61

中的

 -gencode arch=compute_20,code=sm_20 \
-gencode arch=compute_20,code=sm_21 \

去掉即可。

重新clean再编译一边

$ make clean
$ make all -j4
$ make test -j4
$ make runtest -j4


[----------] 2 tests from BilinearFillerTest/1, where TypeParam = double
[ RUN      ] BilinearFillerTest/1.TestFillEven
[       OK ] BilinearFillerTest/1.TestFillEven (5 ms)
[ RUN      ] BilinearFillerTest/1.TestFillOdd
[       OK ] BilinearFillerTest/1.TestFillOdd (8 ms)
[----------] 2 tests from BilinearFillerTest/1 (13 ms total)

[----------] 1 test from SolverTest/1, where TypeParam = caffe::CPUDevice<double>
[ RUN      ] SolverTest/1.TestInitTrainTestNets
[       OK ] SolverTest/1.TestInitTrainTestNets (1 ms)
[----------] 1 test from SolverTest/1 (1 ms total)

[----------] Global test environment tear-down
[==========] 2063 tests from 271 test cases ran. (605597 ms total)
[  PASSED  ] 2063 tests.

如上面所示,说明编译成功了。

8. 配置 pycaffe 接口

在 caffe 根目录的 python 文件夹下,有一个 requirements.txt 的清单文件,上面列出了需要的依赖库,按照这个清单安装就可以了。在安装 scipy 库的时候,需要先安装 fortran 编译器
(gfortran),如果没有这个编译器就会报错,因此,我们可以先安装一下。
a.首先回到 caffe 的根目录,然后执行安装代码:

$ sudo apt-get install gfortran
$cd ./python

b.在终端输入下面一段即可

for req in $(cat requirements.txt); do pip install $req; done

错误:
IOError: [Errno 13] 权限不够: '/home/fc/anaconda2/lib/python2.7/site-packages/leveldb.so'
换root用户重新执行上面的命令

c.安装完成以后,再次回到 caffe 根目录我们可以执行:

$ cd ~/caffe
$ sudo pip install -r python/requirements.txt

问题:
Command "python setup.py egg_info" failed with error code 1 in /tmp/pip-build-Io7ETq/ipython/
You are using pip version 8.1.1, however version 9.0.1 is available.
You should consider upgrading via the 'pip install --upgrade pip' command.

$ sudo python -m pip install --upgrade --force pip
$ sudo pip install setuptools==33.1.1

9.编译 python 接口

$ sudo make pycaffe

10. 配置环境变量,以便 python 调用

$sudo vim ~/.bashrc

加入
export PYTHONPATH=/home/fc/caffe/python:$PYTHONPATH

$ source ~/.bashrc

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 210,978评论 6 490
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 89,954评论 2 384
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 156,623评论 0 345
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,324评论 1 282
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,390评论 5 384
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,741评论 1 289
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,892评论 3 405
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,655评论 0 266
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,104评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,451评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,569评论 1 340
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,254评论 4 328
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,834评论 3 312
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,725评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,950评论 1 264
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,260评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,446评论 2 348