Ubuntu16.04-x64安装caffe2(仅CPU)

Ubuntu16.04-x64安装caffe2(仅CPU)

Caffe2是一个相对比较新的平台,在caffe2的安装过程中可能会出现各种各样的问题,大家保持耐心,总是可以安装好的。

Caffe2的安装过程大家可以参考caffe2官网提供的教程https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=compile

按照官网教程进行安装总归是没错的,但在安装过程中会出现很多问题,如更新源,pip,memory error,虚存等,在以下的安装过程中,会对一出现的问题提供解决方案。

       笔者的环境是Ubuntu16.04-x64,配置如下


1、首先更新一下源,并安装依赖文件,以及pip,最好将3条指令分开进行

sudo apt-get update
sudo apt-get install -y --no-install-recommends \

      build-essential \

      cmake \

      git \

      libgoogle-glog-dev \

      libprotobuf-dev \

      protobuf-compiler \

      python-dev \

      python-pip                         

sudo pip install numpyprotobuf

先update一下


下载python pip等的依赖文件


使用pip下载numpy和protobuf等,在使用pip下载的过程中,有些同学可能会发现,下载速度超级慢,并且可能会出现下载中断等问题。


这是因为这里使用的镜像的问题,可以将镜像改为国内镜像,可以参考http://blog.csdn.net/wukai0909/article/details/62427437

创建一个.pip文件夹,在文件夹内传建一个文件pip.conf

mkdir ~/.pip

cd ~/.pip

vi pip.conf

将下列文字写入文件内,保存并退出

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
[install]
trusted-host=mirrors.aliyun.com

再次执行第三条安装语句,下载速度就非常快了


2、安装libgflags

# for Ubuntu 14.04
sudo apt-get install -y --no-install-recommends libgflags2
# for Ubuntu 16.04
sudo apt-get install -y --no-install-recommends libgflags-dev

这个我已经安装过,所以不会再安装。

3、安装各种依赖文件

 # for both Ubuntu 14.04 and 16.04
sudo apt-get install -y --no-install-recommends \
      libgtest-dev \
      libiomp-dev \
      libleveldb-dev \
      liblmdb-dev \
      libopencv-dev \
      libopenmpi-dev \
      libsnappy-dev \
      openmpi-bin \
      openmpi-doc \
      python-pydot
sudo pip install \
      flask \
      future \
      graphviz \
      hypothesis \
      jupyter \
      matplotlib \
      pydot python-nvd3 \
      pyyaml \
      requests \
      scikit-image \
      scipy \
      setuptools \
      six \
      tornado

安装过程时间比较长,请耐心等待,不过这里一般不会出现错误


但是在使用pip的时候可能会出现错误

Scipy和scikit-image是依赖关系比较复杂的依赖文件,所以常常会在这里出现错误,可以使用这个教程来解决问题https://stackoverflow.com/questions/29466663/memory-error-while-using-pip-install-matplotlib



然后又出现了 command “python setup.py egg-info”failed with error code 1 in /tmp/pip-build*

使用这个教程http://outofmemory.cn/code-snippet/36001/python-pip-python3 安装pip3,并使用pip3重新安装pip安装过的文件

sudo apt-get install python-pip3

sudo pip3 install python3-pip

sudo pip3 install packagename

4、从git下载caffe2,约400M

git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2
make &&cd build && sudo make install

python -c 'from caffe2.python import core' 2>/dev/null&&echo"Success"||echo"Failure"


Git速度比较慢,静静等待

Caffe2下载完成,紧接着make一下,若出现以下内容,则证明安装程序正在进行


还是等待。。。


在进行到76%的时候等待的时间会很长,这往往是由于虚存不足导致的,可以按照下面这个教程解决问题http://blog.csdn.net/yanlai20/article/details/46907353

首先创建swap文件夹

mkdirswap

cd swap

创建文件

 sudo dd if=/dev/zero of=swapfile bs=1024 count=4066544

把生成的文件转换成swap文件

sudomkswap swapfile

激活swap文件

sudo swaponswapfile

修改/etc/fstab,在文件后面加上 /home/username/swap/swapfile   swap    swap   defaults  0  0,保存


完成后可以通过下面的指令查看虚存使用状况

free –m


回到caffe2安装界面,ctrl+c中断目前安装,并重新执行make过程,发现安装可以从76%据需往下执行,直到安装完成,推荐执行make clean后重新make



执行cd build && sudo make install,得到下面的结果


顺利构建,再使用python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

验证安装结果


输出success则安装成功

5、在python中配置caffe2路径

vi ~/.bashrc

追加以下内容,并根据自己的主机名修改“ubuntu”部分

 
export PYTHONPATH=/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/build
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

 


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值