Ubuntu16.04 + Anaconda 安装 caffe2

官网教程在：地址1

1. 下载caffe安装包

  方法1：已经下载过的用户

    更新安装包：

git submodule update

  方法2：从没下载过的用户

   下载安装包，默认下载位置是home底下

git clone --recursive https://github.com/pytorch/pytorch.git

----------------------------------------------------------------------------------------------------- 

2. 安装caffe2

  1. 打开安装包目录

cd pytorch

   2. 安装

     a. 安装 cpu 版本

./scripts/build_anaconda.sh --install-locally

     b. 安装gpu版本(以CUDA 9.0 和 CuDNN 7为例)

./scripts/build_anaconda.sh --install-locally --cuda 9.0 --cudnn 7

---------------------------------------------------------------- 

3. 检验是否安装成功  

  cpu版本:

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

  返回success，即为成功

  gpu版本：

python -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'

  返回具体数字(>0)，即为成功