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