安裝 Caffe for Ubuntu
您必須安裝 Caffe 深度學習架構及相關的套件。Caffe 用於模型訓練及缺陷分類。
程序
-
使用下列指令安裝 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 -
使用下列指令下載 Caffe 原始碼:
wget https://github.com/BVLC/caffe/archive/1.0.zip -
使用下列指令將套件解壓縮並輸入套件目錄:
unzip 1.0.zip
cd ./caffe-1.0 -
使用下列指令複製建立配置檔:
cp Makefile.config.example Makefile.config -
在 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 -
在 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' '{}' \; -
執行下列指令:
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 -
使用下列指令以在 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 部署路徑。 -
開啟
{caffe-installation-path}
目錄中的 makefile,並將參數 NVCCFLAGS 變更為下列設定:
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS) -
在主要的 Caffe 目錄 caffe-1.0 中,使用下列指令開始執行 Caffe 建置及安裝:
make all
make test
make runtest
make pycaffe
make distribute -
將下行新增至 ~/.bashrc Script 中:
export PYTHONPATH="/usr/lib/python2.7:{caffe-installation-path}/caffe-1.0/python:$PYTHONPATH"
其中{caffe-installation-path}
是 Caffe 部署路徑。