Сборка Caffe

Для работы DIGITS требуется сборка Caffe. Официально мы поддерживаем только последние версии из NVIDIA/caffe (NVcaffe), но, вероятно, подойдёт и любая недавняя сборка BVLC/caffe.

Зависимости

Перед сборкой Caffe необходимо собрать Protobuf 3. Следуйте этому руководству, чтобы собрать Protobuf 3.

Для лучшей производительности вам понадобятся:

• Один или несколько графических процессоров NVIDIA (подробности).
• Драйвер NVIDIA (подробности и инструкции по установке).
• CUDA Toolkit (подробности и инструкции по установке).
• cuDNN (страница загрузки).

Установите некоторые зависимости с помощью пакетов Deb:

sudo apt-get install --no-install-recommends build-essential cmake git gfortran libatlas-base-dev libboost-filesystem-dev libboost-python-dev libboost-system-dev libboost-thread-dev libgflags-dev libgoogle-glog-dev libhdf5-serial-dev libleveldb-dev liblmdb-dev libopencv-dev libsnappy-dev python-all-dev python-dev python-h5py python-matplotlib python-numpy python-opencv python-pil python-pip python-pydot python-scipy python-skimage python-sklearn

Скачивание исходного кода

В настоящее время DIGITS совместим с Caffe 0.15.

# пример расположения - можно настроить
export CAFFE_ROOT=~/caffe
git clone https://github.com/NVIDIA/caffe.git $CAFFE_ROOT -b 'caffe-0.15'

Установка переменной среды CAFFE_ROOT поможет DIGITS автоматически обнаружить вашу установку Caffe, но это необязательно.

Пакеты Python

Необходимо установить несколько пакетов PyPI:

sudo pip install -r $CAFFE_ROOT/python/requirements.txt

Если вы столкнулись с ошибками о пропущенных импортах, используйте эту команду для установки пакетов в нужном порядке (см. обсуждение здесь):

cat $CAFFE_ROOT/python/requirements.txt | xargs -n1 sudo pip install

Сборка

Мы рекомендуем использовать CMake для настройки Caffe вместо сборки с использованием Makefile для автоматического обнаружения зависимостей:

cd $CAFFE_ROOT
mkdir build
cd build
cmake ..
make -j"$(nproc)"
make install