Ubuntu16.04安装caffe(gpu版)

继续前面tensorflow(gpu版)、torch后,我们开始安装caffe

若要安装NVIDIA显卡驱动或者安装CUDA+cudnn,参见://www.greatytc.com/p/1baf9848b7d7 (内附下载资源)
若需要安装tensorflow或者torch,参见://www.greatytc.com/p/038c6e073201

1.首先,我们安装一下依赖包

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 libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install git cmake build-essential

2. 安装opencv,注意:我们不安装opencv3,一般用也只用opencv2,没有必要用3

下载opencv版本自选,这里选择下载opencv2.4.13,下载地址:https://opencv.org/releases.html
(1) 将下载好的文件夹解压,进入文件夹 opencv-2.4.13:

cd opencv-2.4.13

(2) 新建一个文件夹用于存放临时文件:

mkdir release

(3) 切换到该临时文件夹:

cd release

(4) 开始编译:

cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local ..

make -j8 //开启线程 按照自己的配置
sudo make install

然后进行环境配置:

(1) 将opencv的库加入到路径,从而让系统可以找到:

sudo gedit /etc/ld.so.conf.d/opencv.conf

(2) 末尾加入/usr/local/lib,保存退出:

sudo ldconfig #使配置生效

sudo gedit /etc/bash.bashrc

末尾加入:

PKG_CONFIG_PATH=$PKG_CONFIG_PATH:/usr/local/lib/pkgconfig
export PKG_CONFIG_PATH

保存退出

sudo source /etc/bash.bashrc #使配置生效

到此为止,opencv的所有工作就做完了

可能有些小伙伴在终端测试时会发现问题:

ImportError: dynamic module does not define init function (PyInit_cv2)

这时候只需要通过pip安装opencv-python就好了:

pip3 install opencv-python

3. 下面我们开始安装caffe

在此确认步骤1中的步骤是否完成(重新输入步骤1中的命令以确认),安装好后方可继续
在自己想安装caffe的位置打开终端:

git clone https://github.com/BVLC/caffe.git

进入 caffe ,将 Makefile.config.example 文件复制一份并更名为 Makefile.config :

cd caffe
sudo cp Makefile.config.example Makefile.config

在 caffe 目录下,修改 Makefile.config 文件:

sudo gedit Makefile.config

修改 Makefile.config 文件内容:
(1).应用 cudnn


 #USE_CUDNN := 1
修改成:
USE_CUDNN := 1

(2).应用 opencv


 #USE_OPENCV := 0
修改为:
USE_OPENCV := 1

(3).使用 python 接口


 #WITH_PYTHON_LAYER := 1
修改为
WITH_PYTHON_LAYER := 1

(4).修改 python 路径:

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

然后修改 caffe 目录下的 Makefile 文件:

sudo gedit Makefile


 NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
替换为:
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)


 LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
改为:
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

然后修改 /usr/local/cuda/include/host_config.h 文件 :

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


 #error-- unsupported GNU version! gcc versions later than 4.9 are not supported!
改为
//#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!

OK ,可以开始编译了,在 caffe 目录下执行 :

make all -j8

这是如果之前的配置或安装出错,那么编译就会出现各种各样的问题,所以前面的步骤一定要细心。

编译成功后可运行测试:

sudo make runtest -j8

4. 最后我们安装 pycaffe接口环境

pycaffe 的安装是为了更方便的去使用 caffe:

cd caffe
sudo make pycaffe -j8

5. 验证

编译 pycaffe 成功后,验证一下是否可以在 python 中导入 caffe 包,首先进入 python 环境:

python

然后导入 caffe :

 >>> import caffe

若不报错则表示 caffe 的 python 接口已正确编译

在终端import caffe成功,但是当使用jupytr notebook时失败的解决办法


如图,出现这种情况时,我们采取一下方式解决:

import os
import sys
caffe_root='/home/haoyev5/caffe/python' #自己的caffe路径
os.chdir(caffe_root)
sys.path.insert(0,caffe_root+'python')

第三行有些是caffe-mast文件夹下的路劲,改下就好了,但是有个弊端就是,每次使用caffe时都必须加入以上几行,略为不便
另外有些说法是这样子的:

import sys
sys.path.append('/home/haoyev5/caffe/python')

我采取了这种方式,无效,大家可以试试

另外,若遇到有关于libpng16.so的错误,如 ImportError: libpng16.so.16: cannot open shared object file: No such file or directory,按照https://blog.csdn.net/pkokocl/article/details/78756422下载安装就好了。

感谢:https://blog.csdn.net/yhaolpz/article/details/71375762

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

推荐阅读更多精彩内容