Инструкции за сваляне и инсталиране рамките на дълбоко изучаване кафе в NVIDIA
Инструкции за сваляне и инсталиране на Caffe да се движат по графичните чипове на NVIDIA
Системни изисквания да тече CAFFE
Системни изисквания за Caffe с подкрепа за ускоряване на графичния процесор:
- 64-битова операционна система Linux (Това ръководство е предназначено за операционната система Ubuntu 14.04)
- NVIDIA ® ® CUDA NVIDIA CUDA 7.5 (за графичния процесор с NVIDIA Pascal ™ архитектура изисква CUDA 8.0)
- cuDNN v5.1
- Имате нужда от NVIDIA GPU изчислителна способност 3.0 или по-късно.
Инструкции за сваляне и инсталиране на Caffe
Стъпка 1. Инсталирайте CUDA
За да използвате Caffe на графичния процесор NVIDIA първо задайте CUDA Toolkit.
Стъпка 2. Поставете cuDNN
След като инсталирате CUDA Toolkit изтегляне cuDNN v5.1 библиотека за Linux на (имайте предвид, че ще трябва да се запишат в програмата за разработчици Computing на програмата ускорен).
След изтеглянето, разархивирайте библиотечни файлове и да ги копирате в набора от инструменти за директория CUDA (/ ЮЕсАр / местни / CUDA /):
$ Sudo катран -xvf cudnn-8,0-Linux-x64-v5.1.tgz -C / ЮЕсАр / местни
Стъпка 3. Задайте зависимост
Caffe е в зависимост от няколко библиотеки, които трябва да бъдат на разположение във вашата система чрез управление на пакетите.
Операционната система Ubuntu 14.04 следните команди ви позволяват да инсталирате необходимите библиотеки:
$ Ап-да инсталирате libprotobuf-Dev libleveldb-Dev libsnappy-Dev libopencv-Dev libhdf5-сериен Дев Protobuf-компилатора libgflags-Dev libgoogle-Вземи в-Dev liblmdb-Dev libatlas база-Dev Git
$ Ап-да инсталирате --no инсталиране-препоръчва libboost-ол-Dev
Стъпка 4. Монтирайте NCCL
NVIDIA NCCL Caffe се изисква да се движат по множествена GPU. Инсталиране NCCL изпълнете следните команди:
$ Sudo направи инсталирате -j4
Библиотеки и заглавни файлове NCCL ще се инсталира в / ЮЕсАр / местни / ИЪ и / ЮЕсАр / местни / включва.
Стъпка 5. Комплект Caffe
$ Tar -zxf v0.15.9.tar.gz
$ Cp Makefile.config.example Makefile.config
Разкоментирайте линия USE_CUDNN: = 1. Това позволява cuDNN ускорение.
Премахнете линията USE_NCCL: = 1. Това дава възможност NCCL която е необходима за стартирането на Caffe на множество графични процесори.
Запазване и затворете файла. Вече сте готови да се съберат Caffe.
След командата Caffe бинарни файлове ще бъдат на разположение в изграждане / инструменти / кафе.
Подготовка с данни за изображенията
Стъпка 6: Изтеглете допълнителните данни
Стъпка 7: Създаване на база данни
Промяна на променливи TRAIN_DATA_ROOT и VAL_DATA_ROOT до директорията, където сте извлечена от изображението.
Комплект функция преоразмерите = вярно, за да промените размера на изображенията, преди да добавите към базата данни.
Запазване и затворете файла. Сега се създаде база данни на изображението със следните команди:
След това създайте необходимата средна файл: