ubuntu怎么安装caffe2

ubuntu怎么安装caffe2

Caffe2的安装相比于caffe，在安装的时候更加简便，略去了Makefile.config的各种配置。本经验的步骤中包含两个可选项，即“可选GPU支持（步骤2）”和“可选依赖库（步骤3）”，如果大家不安装它们，可以跳过这两个步骤；建议大家安装它们，它们有太多的好处了。现在就让我们看看怎么在ubuntu中安装Caffe2吧。

命令如下。

（1）sudo apt-get update

（2）sudo apt-get install -y --no-install-recommends \

build-essential \

cmake \

git \

libgoogle-glog-dev \

libprotobuf-dev \

protobuf-compiler \

python-dev \

python-pip

（3）sudo pip install numpy protobuf

概述。如果您打算使用GPU而不是CPU，那么您应该安装NVIDIA CUDA 8和cuDNN v5.1或v6.0，这是GPU加速的深度神经网络的原生类型库。

首先更新您的显卡驱动程序（NVIDIA CUDA 8）！ 否则您可能会遇到各种难以诊断的错误。

1、适用Ubuntu 16.04的命令行。

（1）sudo apt-get update && sudo apt-get install wget -y --no-install-recommends

（2）wget "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb"

（3）sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb

（4）sudo apt-get update

（5）sudo apt-get install cuda

2、适用Ubuntu 14.04的命令行。

（1）sudo apt-get update && sudo apt-get install wget -y --no-install-recommends

（2）wget "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.61-1_amd64.deb"

（3）sudo dpkg -i cuda-repo-ubuntu1404_8.0.61-1_amd64.deb

（4）sudo apt-get update

（5）sudo apt-get install cuda

安装cuDNN（适用所有Ubuntu版本）。

（1）CUDNN_URL="http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz"

（2）wget ${CUDNN_URL}

（3）sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local

（4）rm cudnn-8.0-linux-x64-v5.1.tgz && sudo ldconfig

libgflags库安装。注意：libgflags2是用于Ubuntu 14.04；libgflags-dev是用于Ubuntu 16.04。这两个不能搞混，否则安装会报错。

(1)适用Ubuntu 14.04的命令行。

sudo apt-get install -y --no-install-recommends libgflags2

(2)适用Ubuntu 16.04的命令行。

sudo apt-get install -y --no-install-recommends libgflags-dev

安装其他可选依赖库（适合Ubuntu 14.04和16.04）。

（1）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

（2）sudo pip install \

flask \

future \

graphviz \

hypothesis \

jupyter \

matplotlib \

pydot python-nvd3 \

pyyaml \

requests \

scikit-image \

scipy \

setuptools \

six \

tornado

命令如下。

（1）git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2

（2）make && cd build && sudo make install

更改配置文件。

（1）打开配置文件的命令行：sudo vim /etc/profile

（2）然后在配置文件的最后添加如下代码：

export PYTHONPATH=/usr/local:$PYTHONPATH

export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/build

export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

（3）注意：/home/ubuntu/caffe2/build这部分要根据你的实际目录来填写（export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe2/build）。

使修改生效。重新启动使修改配置文件生效。

检查设置是否正确。如果设置正确的话，执行命令以后，会有如下图这样的输出。

（1）命令：echo $PYTHONPATH

（2）命令：echo $LD_LIBRARY_PATH

caffe2测试。运行下面的命令来测试你的caffe2是否成功。如果输出“Success”，表示成功；如果输出“Failure”，表示失败。

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

查看caffe2测试的失败原因。如果上述测试失败了，请输入下面命令 ，查看一下失败的原因。

（1）python -c 'from caffe2.python import core'

GPU测试。运行下面的命令来测试你的GPU安装是否成功。

（1）python -m caffe2.python.operator_test.relu_op_test