当前位置:首页 > 芯闻号 > 充电吧
[导读]Ubuntu16.04-x64安装caffe2(仅CPU)Caffe2是一个相对比较新的平台,在caffe2的安装过程中可能会出现各种各样的问题,大家保持耐心,总是可以安装好的。Caffe2的安装过程

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

 


本站声明: 本文章由作者或相关机构授权发布,目的在于传递更多信息,并不代表本站赞同其观点,本站亦不保证或承诺内容真实性等。需要转载请联系该专栏作者,如若文章内容侵犯您的权益,请及时联系本站删除( 邮箱:macysun@21ic.com )。
换一批
延伸阅读
关闭