最近做的课题需要阅读Caffe的代码,就抽时间配一下。发现官方教程是基于Python2.7的,做死尝试了一次Python3.6没有成功(以后有空再研究),所以就先用Python2.7.15 + Protobuf2.6.1配了一下。 因为担心Caffe会影响Miniconda3的其他环境,这里为Caffe单独创建一个沙盒环境。
前置要求是你需要成功配置GPU或者CPU,我使用的是Miniconda3 + CUDA 9.0 + CuDNN 7.1.4,具体可以参考博文
1. 创建Caffe沙盒
这里我使用Protobuf2.6.1
的原因在于OpenPose是不兼容Miniconda3新的Protobuf Compiler。
conda create --name caffe-py2.7.15-pr2.6.1 python=2.7.15
2. 下载Caffe
进入miniconda3/envs/caffe-py2.7.15-pr2.6.1
, 执行以下命令
cd ~/miniconda3/envs/caffe-py2.7.15-pr2.6.1/
git clone https://github.com/BVLC/caffe.git
如果你需要某个commit
版本的Caffe,执行
cd caffe/
git checkout commit
如果你需要编译OpenPose(今天是2018.11.19,该贡献人员还没解决因最新版Caffe增加Layer导致与OpenPose不兼容的问题,后期解决后就不需要这么做了),当前的commit
为f019d0dfe86f49d1140961f8c7dec22130c83154
。
小组的同学需要执行
cd caffe/
git checkout f019d0dfe86f49d1140961f8c7dec22130c83154
3. 安装依赖
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install -y build-essential cmake git pkg-config
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-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 libopencv-dev libhdf5-serial-dev
sudo apt-get install -y build-essential cmake git libgtk2.0-dev pkg-config python-dev python-numpy libdc1394-22 libdc1394-22-dev libjpeg-dev libpng12-dev libtiff5-dev libjasper-dev libavcodec-dev libavformat-dev libswscale-dev libxine2-dev libgstreamer0.10-dev libgstreamer-plugins-base0.10-dev libv4l-dev libtbb-dev libqt4-dev libfaac-dev libmp3lame-dev libopencore-amrnb-dev libopencore-amrwb-dev libtheora-dev libvorbis-dev libxvidcore-dev x264 v4l-utils unzip
4. 激活caffe环境并安装opencv
source activate caffe-py2.7.15-pr2.6.1
conda install opencv
5. Protobuf版本选择
请注意这一步非常非常重要!!!
编译Caffe时Protobuf版本的选择也是有讲究的(如果你需要其他特定版本的库,具体做法和这个类似)。拿我的库举个例子:Miniconda3的Protobuf版本是3.5.2,而通过apt-get 安装的Protobuf版本是2.6.1。
Github上有些开源项目使用的是较低版本(如2.6.1)的Protobuf,你用高版本(如3.5.2)的Protobuf编译后的Caffe去配置他们的项目肯定会失败的。编译Caffe所使用的依赖库版本在执行7.1
后终端会给出提示信息。
注意:5.1
和5.2
只需要执行一个即可,5.3
是教大家如何安装和删除protobuf的,没有需求跳过即可。小组成员请选择5.2
执行
5.1 使用Conda的Protobuf
只要Miniconda3的路径在bashrc
,而且你通过Conda安装了libprotobuf和protobuf这两个库,那么直接编译Caffe即可。对我的机器而言,这时候编译Caffe是基于Protobuf3.5.2的。
source activate caffe-py2.7.15-pr2.6.1
conda install protobuf
5.2 使用自己的版本
首先你要保证当前的沙盒环境中没有libprotobuf和protobuf这两个库(激活当前的caffe-py2.7.15-pr2.6.1
,然后conda list
查看是否有这两个库,有的库需要删掉),然后注释掉Miniconda3在bashrc
的环境变量,接下来就可以直接使用系统的库去编译Caffe了。当然如果你想使用一些特定版本(非系统库)的库去编译,你还需要将它的路径导入bashrc
,然后再编译。
vim ~/.bashrc
注释掉Miniconda3的环境变量
sourc