Ubuntu 16.04 在Conda沙盒环境下安装Caffe(Python2.7.15 + Protobuf2.6.1 + GPU)

最近做的课题需要阅读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不兼容的问题,后期解决后就不需要这么做了),当前的commitf019d0dfe86f49d1140961f8c7dec22130c83154

小组的同学需要执行

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.15.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
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值