Caffe 安装踩坑记录
很长时间没有安装 Caffe 了,这次记录下自己安装的过程,也为可能和我遇到相同问题的小伙伴提供一点点帮助。
首先说明环境:
Ubuntu16.04 +CUDA10+cudnn7.3+1080Ti
特别说明下,因为本地环境中已经装好了 CUDA10+cudnn7.3,,这里就不详细说明安装步骤了。
同时,本文参考了这位博友的cuda安装博文
我的安装过程分为 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
//修改的地方如下
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。
评论