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 -DWITH_GTK_2_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
CUDA_ARCH := -gencode arch=compute_61,code=sm_61
PYTHON_LIBRARIES := boost_python3 python3.5m
PYTHON_INCLUDE := /usr/include/python3.5m \
/usr/lib/python3.5/dist-packages/numpy/core/include
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/
//修改Makefile文件
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
//编译Caffe
sudo make all -j8
这里特别说一下,编译过程中出现的一个小BUG
编译时,出现了如下图的错误,
可查看git issue
最后,编译成功的情况下,会得到如下的结果
结语
以上就是此次踩坑的过程,希望可以对遇到相似问题的小伙伴帮助。在编译的过程中,可能会出现一些奇怪的问题,愿大家可以冷静分析,多多Google。