转载请声明,谢谢。
Facebook终于开源了caffe2,参照Tutorials,完成caffe2安装。
参考:https://caffe2.ai/docs/getting-started.html?platform=linux&configuration=compile
特别注意
目前caffe2 Github更新频繁,经常出现build error状态,建议在如下状态进行git操作。
安装依赖
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 numpy protobuf
GPU支持 (可选)
如果打算使用GPU训练模型,需要安装CUDA和CUDNN。
需要安装N卡闭源驱动!
在ubuntu16.04上安装CUDA:
sudo apt-get update && sudo apt-get install wget -y --no-install-recommends
wget "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_8.0.61-1_amd64.deb"
sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
安装CUDNN:
CUDNN_URL="http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz"
wget ${CUDNN_URL}
sudo tar -xzf cudnn-8.0-linux-x64-v5.1.tgz -C /usr/local
rm cudnn-8.0-linux-x64-v5.1.tgz && sudo ldconfig
安装可选依赖
sudo apt-get install -y --no-install-recommends \
libgflags-dev \
libgtest-dev \
libiomp-dev \
libleveldb-dev \
liblmdb-dev \
libopencv-dev \
libopenmpi-dev \
libsnappy-dev \
openmpi-bin \
openmpi-doc \
python-pydot
sudo pip install \
flask \
graphviz \
hypothesis \
jupyter \
matplotlib \
pydot python-nvd3 \
pyyaml \
requests \
scikit-image \
scipy \
setuptools \
tornado
下载源码
git clone --recursive https://github.com/caffe2/caffe2.git
源码较大,加上子模块有160兆。
编译和安装
cd caffe2
make && cd build && sudo make install
测试
python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"
上述测试输出Success即表示安装成功
python -m caffe2.python.operator_test.relu_op_test
上述测试中,如果没有GPU支持,在开头会有警告,但是不影响运行。
环境配置
最后还需要配置环境,以便在下次启动终端能够正常运行。
cd ~/
gedit .bashrc
在最后加入:
export PYTHONPATH=/usr/local:$PYTHONPATH
export PYTHONPATH=$PYTHONPATH:/path/to/caffe2/build
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
注意修改/path/to/caffe2指向caffe2目录。
至此caffe2安装完毕。