Installation de Caffe pour Ubuntu

Vous devez installer l'infrastructure d'apprentissage en profondeur Caffe et les packs associés. Caffe est utilisé pour l'entraînement des modèles et la classification des défauts.

1. Installez les packs requis pour Caffe avec les commandes suivantes :
   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. Téléchargez le code source de Caffe avec la commande suivante :
   wget https://github.com/BVLC/caffe/archive/1.0.zip
3. Décompressez le pack et placez-vous dans le répertoire du pack avec les commandes suivantes :
   unzip 1.0.zip
   cd ./caffe-1.0
4. Effectuez une copie du fichier de configuration make avec la commande suivante :
   cp Makefile.config.example Makefile.config
5. Ajoutez les variables suivantes dans le fichier 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. Dans le répertoire caffe-1.0, exécutez la commande suivante :
   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. Exécutez les commandes suivantes :
   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. Installez les packages Python requis dans le répertoire caffe-1.0/python avec les commandes suivantes :
   cd {chemin-installation-caffe}/caffe-1.0/python
   for req in $(cat requirements.txt); do sudo -H pip install $req --upgrade; done
   où {chemin-installation-caffe} est le chemin de déploiement de Caffe.
9. Ouvrez le fichier makefile dans le répertoire {chemin-installation-caffe} et remplacez le paramètre NVCCFLAGS comme suit :
   NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
10. Dans le répertoire Caffe principal caffe-1.0, commencez la génération et l'installation de Caffe avec les commandes suivantes :
    make all
    make test
    make runtest
    make pycaffe
    make distribute
11. Ajoutez la ligne suivante dans le script ~/.bashrc :
    export PYTHONPATH="/usr/lib/python2.7:{chemin-installation-caffe}/caffe-1.0/python:$PYTHONPATH"
    où {chemin-installation-caffe} est le chemin de déploiement de Caffe.