[PYTHON] Installieren Sie Caffe unter Ubuntu 14.04 (GPU)

Über den Installationsvorgang von Caffe. Das, was ich vorher geschrieben habe, war ziemlich subtil, also habe ich es umgeschrieben. (Ich bin nur Chainer und ich benutze kein Caffe.) Vorherige: Caffe unter Ubuntu 14.04 installieren (CPU-Modus)

Ich habe es auf Ubuntu 14.04 versucht. Nun, es ist das gleiche wie auf der offiziellen Website. Ich werde ein Memo schreiben.

CUDA-Installation

#Installieren Sie die neueste Version
$ wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_7.5-18_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1404_7.5-18_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

cuDNN Es scheint, dass es schneller sein wird, wenn cuDNN eingeführt wird. (Registrierung des Nvidia-Entwicklers erforderlich) Sie müssen es nicht eingeben.

#Registrieren Sie sich und laden Sie es von Ihrem Browser herunter
$ tar xvf cudnn-7.5-linux-x64-v5.0-rc-tgz
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp cuda/lib64/* /usr/local/cuda/lib64

Abhängigkeitsinstallation

$ sudo apt-get install libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler

Laden Sie Caffe herunter

#Wenn Git nicht enthalten ist
$ sudo apt-get install git

#Stellen Sie es direkt unter Ihr Zuhause. Wenn Sie verstehen, gehen Sie bitte, wohin Sie wollen.
$ cd
$ git clone https://github.com/BVLC/caffe.git

Einstellungen für Makefile.config

Wenn Sie cudnn nicht verwenden, können Sie es so lassen, wie es ist

$ cd caffe
$ cp Makefile.config.example Makefile.config

Kommentieren Sie Zeile 5 aus, wenn Sie cnDNN verwenden

Makefile.config

# USE_CUDNN := 1
↓
USE_CUDNN := 1

kompilieren

Die parallele Verarbeitung kann mit $ make -j4 all erfolgen. Die Anzahl ist die Anzahl der CPU-Kerne. Es braucht Zeit, daher wird empfohlen, dies mit voller Leistung zu tun.

$ make all
$ make test
$ make runtest

Wenn alle Tests bestanden sind, ist der Aufbau der Kaffee-Umgebung abgeschlossen. Wenn dies fehlschlägt, lösen Sie das Problem, machen Sie "$ make clean" und beginnen Sie von "$ make all".

Erstellen einer Umgebung für die Verwendung von Caffe mit Python

Python-System

$ sudo apt-get install python-dev python-pip python-numpy python-skimage gfortran

Installation der benötigten Bibliotheken mit pip

$ sudo pip install -r ~/caffe/python/requirements.txt

kompilieren

$ make pycaffe

Durchlaufen. (Bash Beispiel)

$ echo 'export PYTHONPATH=~/caffe/python/:$PYTHONPATH' >> ~/.bashrc
#Reflektieren
$ source ~/.bashrc

später

$ python
>>> import caffe

OK, wenn Kaffee mit geladen ist.