À propos de la procédure d'installation de Caffe. Celui que j'ai écrit avant était assez subtil, alors je l'ai réécrit. (Je suis juste Chainer et je n'utilise pas Caffe.) Précédent: Installer Caffe sur Ubuntu 14.04 (mode CPU)
Je l'ai essayé sur ubuntu 14.04. Eh bien, c'est la même chose que le site officiel. J'écrirai une note.
#Installez la dernière 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 Il semble que ce sera plus rapide si cuDNN est introduit. (Inscription développeur Nvidia requise) Vous n'êtes pas obligé de le mettre.
#Inscrivez-vous et téléchargez depuis votre navigateur
$ 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
$ sudo apt-get install libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
#Si git n'est pas inclus
$ sudo apt-get install git
#Placez-le directement sous la maison. Si vous comprenez, allez où vous voulez.
$ cd
$ git clone https://github.com/BVLC/caffe.git
Si vous n'utilisez pas cudnn, vous pouvez le laisser tel quel
$ cd caffe
$ cp Makefile.config.example Makefile.config
Décommentez la ligne 5 si vous utilisez cnDNN
Makefile.config
# USE_CUDNN := 1
↓
USE_CUDNN := 1
Le traitement parallèle peut être effectué en faisant $ make -j4 all. Le nombre correspond au nombre de cœurs de processeur.
Cela prend du temps, il est donc recommandé de le faire à pleine puissance.
$ make all
$ make test
$ make runtest
Si tous les tests réussissent, la construction de l'environnement caffe est terminée.
Si cela échoue, résolvez le problème, puis faites $ make clean et recommencez à partir de $ make all.
système python
$ sudo apt-get install python-dev python-pip python-numpy python-skimage gfortran
Installation des bibliothèques requises avec pip
$ sudo pip install -r ~/caffe/python/requirements.txt
compiler
$ make pycaffe
Traverser. (Exemple Bash)
$ echo 'export PYTHONPATH=~/caffe/python/:$PYTHONPATH' >> ~/.bashrc
#Refléter
$ source ~/.bashrc
plus tard
$ python
>>> import caffe
OK si caffe est chargé avec.
Recommended Posts