Instrucțiuni pentru descărcarea și instalarea cadrul caffe învățare profundă, nvidia

Instrucțiuni pentru descărcarea și instalarea Caffe pentru a rula pe NVIDIA GPU

Cerințe de sistem pentru a rula CAFFE

Cerințe de sistem pentru Caffe cu suport pentru accelerare pe GPU:

  • sistem de operare pe 64 de biți Linux (Acest manual este proiectat pentru sistemul de operare Ubuntu 14.04)
  • NVIDIA ® CUDA ® NVIDIA CUDA 7.5 (pentru GPU cu arhitectura NVIDIA Pascal ™ necesită CUDA 8.0)
  • cuDNN v5.1
  • Veți avea nevoie de un NVIDIA GPU capacitate de calcul 3.0 sau o versiune ulterioară.

Instrucțiuni pentru descărcarea și instalarea Caffe

Pasul 1. Instalați CUDA

Pentru a utiliza Caffè pe GPU NVIDIA în primul rând set CUDA Toolkit.

Pasul 2. Instalați cuDNN

După ce instalați CUDA Toolkit descărcare cuDNN v5.1 Biblioteca pentru Linux (rețineți că va trebui să se înscrie în program accelerat Programul Developer Computing).

După descărcarea, dezarhivați fișierele de bibliotecă și să le copiați în directorul Toolkit CUDA (/ usr / local / CUDA /):

$ Sudo tar -xvf cudnn-8.0-linux-x64-v5.1.tgz -C / usr / local

Pasul 3. Setați în funcție

Caffe este în funcție de mai multe biblioteci, care ar trebui să fie disponibile în sistemul dvs. prin managerul de pachete.

Sistemul de operare Ubuntu 14.04 următoarele comenzi vă permit să instalați bibliotecile necesare:

$ Sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev libgflags-dev Protobuf-compilator libgoogle-glog-dev liblmdb-dev git libatlas-base-dev

$ Sudo apt-get install --no-install-recomanda libboost-all-dev

Pasul 4. Instalați NCCL

NVIDIA NCCL Caffe este necesar pentru a rula pe GPU multiple. Instalarea NCCL executați următoarele comenzi:

$ Sudo make install -j4

Bibliotecile și antet fișiere NCCL vor fi instalate în / usr / local / lib și / usr / local / include.

Pasul 5. Set Caffe

$ Tar -zxf v0.15.9.tar.gz

$ Cp Makefile.config.example Makefile.config

Decomentați linia USE_CUDNN: = 1. Acest lucru permite accelerarea cuDNN.

Decomentați linia USE_NCCL: = 1. Acest lucru permite NCCL care este necesar pentru a rula pe mai multe unități de procesare grafică Caffe.

Salvați și închideți fișierul. Acum sunteți gata să compileze Caffe.

După comanda binaries Caffe va fi disponibil în build / unelte / caffe.

Pregătirea bazei de date imagine

Pasul 6: Descărcați datele auxiliare

Pasul 7: Crearea unei baze de date

Schimbarea TRAIN_DATA_ROOT variabile și VAL_DATA_ROOT la directorul în care ați extras imaginea.

Set REDIMENSIONARE funcția = true, pentru a redimensiona imaginile înainte de a adăuga la baza de date.

Salvați și închideți fișierul. Acum, creați baza de date de imagine cu următoarele comenzi:

Apoi, creați fișierul mediu necesar: