Caffe2 需要 protobuf、atlas、glog、gtest、limdb、leveldb、snappy、OpenMP、OpenCV、pthread-stubs、cmake、python-protobuf 以及 numpy。
对于 GPU 加速,CUDA 和 cuDNN 是必须的。支持最新版本的 cuDNN (5.1.10)。
下面列出了在 Ubuntu 16.04 系统上安装依赖项的示例过程。
sudo apt-get install libprotobuf-dev protobuf-compiler libatlas-base-dev libgoogle-glog-dev libgtest-dev liblmdb-dev libleveldb-dev libsnappy-dev python-dev python-pip libiomp-dev libopencv-dev libpthread-stubs0-dev cmake python-protobuf git
sudo pip install numpy [matplotlib ipython (如果还想使用 ipython notebook)]