Caffe Installation Tips

Install OpenCV 3.3 or later with Python3 bindings

Accuracy Checker uses OpenCV library for image processing. You can miss this step if you are using OpenCV from OpenVINO toolkit.

sudo apt-get install libopencv-dev
pip install opencv-python

Install Caffe with Python3 bindings

  • Clone repository:
git clone https://github.com/BVLC/caffe.git
cd caffe
  • Install Caffe dependencies:
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler libgflags-dev libgoogle-glog-dev liblmdb-dev
sudo apt-get install --no-install-recommends libboost-all-dev
pip install -r python/requirements.txt
pip install matplotlib
  • Build

If you need CPU only version of caffe add -DCPU_ONLY=ON to cmake command.

mkdir build && cd build
cmake -DCMAKE_BUILD_TYPE=Release -DCMAKE_INSTALL_PREFIX=<caffe/install/dir> -Dpython_version=3 -DBLAS=open ..
make
sudo make install
  • Copy Python library to your python installation.
cp -r ../python/caffe $VIRTUAL_ENV/lib/python3.5/site-packages
cp --remove-destination lib/_caffe.so $VIRTUAL_ENV/lib/python3.5/site-packages/caffe

Check your installation

You can test prerequisites with the following command. If it does not fail, then you are installed prerequisites correctly:

python3 -c 'import caffe, cv2'