写点什么

Caffe 安装踩坑记录

用户头像
Dreamer
关注
发布于: 2020 年 11 月 02 日
Caffe 安装踩坑记录

很长时间没有安装 Caffe 了,这次记录下自己安装的过程,也为可能和我遇到相同问题的小伙伴提供一点点帮助。



首先说明环境:

Ubuntu16.04 +CUDA10+cudnn7.3+1080Ti


特别说明下,因为本地环境中已经装好了 CUDA10+cudnn7.3,,这里就不详细说明安装步骤了。

同时,本文参考了这位博友的cuda安装博文

opencv 安装




我的安装过程分为 3 个步骤:


  • 1.安装依赖包

  • 2.安装 openCV3

  • 3.安装,编译 Caffe


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

//gflags 头文件,这是一个 google 的日志库

sudo apt-get install libgflags-dev

sudo apt-get install lingoogle-glog-dev

//lmdb 头文件

sudo apt-get install liblmdb-dev

//lcblas,latlas 依赖库

sudo apt-get install libatlas-base-dev


2.安装 openCV3


首先在OpenCV Git仓库找到对应版本的 SOUCE CODE,


//安装 cmake

sudo apt-get install cmake

//进入 openCV 下载的路径下,解压

tar -xzvf opencv-3.4.3.tar.gz

//在上一级目录下,先建 build,编译 openCV

cd ..

mkdir build

sudo cmake ../opencv-3.4.3 -DWITHGTK2_X=ON -DCMAKE_INSTALL_PREFIX=/usr/local

sudo make all -j4

sudo make install


3.安装,编译 Caffe


//下载 Caffe 源码

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

//复制 Makefile.config.example

cp Makefile.config.example Makefile.config

//编辑 Makefile.config

gedit Makefile.config

Google

//修改的地方如下

USE_CUDNN := 1

USE_OPENCV := 0

OPENCV_VERSION := 3

CUDAARCH := -gencode arch=compute61,code=sm_61

PYTHONLIBRARIES := boostpython3 python3.5m

PYTHON_INCLUDE := /usr/include/python3.5m \

/usr/lib/python3.5/dist-packages/numpy/core/include

INCLUDEDIRS := $(PYTHONINCLUDE) /usr/local/include /usr/include/hdf5/serial

LIBRARYDIRS := $(PYTHONLIB) /usr/local/lib /usr/lib /usr/lib/x8664-linux-gnu /usr/lib/x8664-linux-gnu/hdf5/serial/

//修改 Makefile 文件

LIBRARIES += glog gflags protobuf boostsystem boostfilesystem m

LIBRARIES += glog gflags protobuf boostsystem boostfilesystem m hdf5serialhl hdf5_serial

//编译 Caffe

sudo make all -j8


这里特别说一下,编译过程中出现的一个小 BUG

编译时,出现了如下图的错误

通过查找资料发现了解决方案,可查看git issue


最后,编译成功的情况下,会得到如下的结果


结语


以上就是此次踩坑的过程,希望可以对遇到相似问题的小伙伴帮助。在编译的过程中,可能会出现一些奇怪的问题,愿大家可以冷静分析,多多 Google。




用户头像

Dreamer

关注

一个不想做搜索的NLPer不是一个好的CVer 2019.12.18 加入

还未添加个人简介

评论

发布
暂无评论
Caffe 安装踩坑记录