安裝 Caffe for Ubuntu

您必須安裝 Caffe 深度學習架構及相關的套件。Caffe 用於模型訓練及缺陷分類。

程序

  1. 使用下列指令安裝 Caffe 所需的套件：
    sudo apt-get update
    sudo apt-get upgrade
    sudo apt-get install -y build-essential cmake git pkg-config
    sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
    sudo apt-get install -y libatlas-base-dev libjasper-dev
    sudo apt-get install -y --no-install-recommends libboost-all-dev
    sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev
    sudo apt-get install -y python-pip
    sudo apt-get install -y python-dev
    sudo apt-get install -y python-numpy python-scipy
    sudo apt-get install -y libopencv-dev
    sudo pip install opencv-python
    sudo pip install flask_httpauth
    sudo pip install gevent
    sudo pip install pyinotify
    sudo pip install tornado
  2. 使用下列指令下載 Caffe 原始碼：
    wget https://github.com/BVLC/caffe/archive/1.0.zip
  3. 使用下列指令將套件解壓縮並輸入套件目錄：
    unzip 1.0.zip
    cd ./caffe-1.0
  4. 使用下列指令複製建立配置檔：
    cp Makefile.config.example Makefile.config
  5. 在 Makefile.config 檔中新增下列變數：
    USE_CUDNN := 1
    CUDA_DIR := /usr/local/cuda
    PYTHON_INCLUDE := /usr/include/python2.7 \
    /usr/lib/python2.7/dist-packages/numpy/core/include
    PYTHON_LIB := /usr/lib/x86_64-linux-gnu
    WITH_PYTHON_LAYER := 1
    INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include \
    /usr/include/hdf5/serial
    LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib \
    /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
  6. 在 caffe-1.0 目錄中，執行下列指令：
    find . -type f -exec sed -i -e 's^"hdf5.h"^"hdf5/serial/hdf5.h"^g' -e 's^"hdf5_hl.h"^"hdf5/serial/hdf5_hl.h"^g' '{}' \;
  7. 執行下列指令：
    cd /usr/lib/x86_64-linux-gnu
    sudo ln -s libhdf5_serial.so.10.1.0 libhdf5.so
    sudo ln -s libhdf5_serial_hl.so.10.0.2 libhdf5_hl.so
  8. 使用下列指令以在 caffe-1.0/python 目錄中安裝必要的 Python 套件：
    cd {caffe-installation-path}/caffe-1.0/python
    for req in $(cat requirements.txt); do sudo -H pip install $req --upgrade; done
    其中 {caffe-installation-path} 是 Caffe 部署路徑。
  9. 開啟 {caffe-installation-path} 目錄中的 makefile，並將參數 NVCCFLAGS 變更為下列設定：
    NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
  10. 在主要的 Caffe 目錄 caffe-1.0 中，使用下列指令開始執行 Caffe 建置及安裝：
    make all
    make test
    make runtest
    make pycaffe
    make distribute
  11. 將下行新增至 ~/.bashrc Script 中：
    export PYTHONPATH="/usr/lib/python2.7:{caffe-installation-path}/caffe-1.0/python:$PYTHONPATH"
    其中 {caffe-installation-path} 是 Caffe 部署路徑。