Instalación de Caffe para Ubuntu

Debe instalar la infraestructura de aprendizaje profundo de Caffe y los paquetes relacionados. Caffe se utiliza para la formación de modelos y la clasificación de defectos.

1. Utilice los mandatos siguientes para instalar los paquetes necesarios para 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. Utilice el mandato siguiente para descargar el código fuente de Caffe:
   wget https://github.com/BVLC/caffe/archive/1.0.zip
3. Utilice los mandatos siguientes para desempaquetar el paquete y especificar el directorio del paquete:
   unzip 1.0.zip
   cd ./caffe-1.0
4. Utilice el mandato siguiente para realizar una copia del archivo de configuración make:
   cp Makefile.config.example Makefile.config
5. Añada las variables siguientes en el archivo 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. En el directorio caffe-1.0, ejecute el mandato siguiente:
   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. Ejecute los mandatos siguientes:
   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. Instale los paquetes Python necesarios en el directorio caffe-1.0/python mediante los mandatos siguientes:
   cd {caffe-installation-path}/caffe-1.0/python
   for req in $(cat requirements.txt); do sudo -H pip install $req --upgrade; done
   donde {ruta_instalación_caffe} es la ruta de despliegue de Caffe.
9. Abra el archivo make en el directorio {ruta_instalación_caffe} y cambie el parámetro NVCCFLAGS al valor siguiente:
   NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)
10. En el directorio principal de Caffe caffe-1.0, utilice los mandatos siguientes para iniciar la creación y la instalación de Caffe:
    make all
    make test
    make runtest
    make pycaffe
    make distribute
11. Añada la línea siguiente al script ~/.bashrc:
    export PYTHONPATH="/usr/lib/python2.7:{caffe-installation-path}/caffe-1.0/python:$PYTHONPATH"
    donde {ruta_instalación_caffe} es la ruta de despliegue de Caffe.