Comment installer DLIB avec 2020 / CUDA activé

Aperçu

J'ai réussi à installer dlib avec CUDA et cuDNN activés, donc je l'ai enregistré sous forme de mémorandum. La méthode d'installation diffère selon la version de dlib et CUDA peut ne pas être activé même si vous suivez l'article sur le net. La méthode d'installation change à chaque fois que dlib est mis à jour, donc si cela ne fonctionne pas, vérifiez dlib official How to compile. La procédure suivante est la méthode d'installation à partir du 6 juillet 2020

environnement

Installation

Installez CUDA, cuDNN

Si vous n'avez pas encore installé CUDA et cuDNN, installez d'abord les deux.

##Installez d'abord CUDA
$ sudo apt update -y
$ sudo apt upgrade -y
$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
$ sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-ubuntu1804-10-2-local-10.2.89-440.33.01_1.0-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-10-2-local-10.2.89-440.33.01/7fa2af80.pub
$ sudo apt-get -y update
$ sudo apt-get -y install cuda
$ echo "export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}" >> ~/.bashrc
$ echo "export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}" >> ~/.bashrc

Redémarrez ici Après le démarrage, confirmez que CUDA est installé avec la commande suivante

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_19:24:38_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89

Si vous pouvez confirmer, l'installation de CUDA est terminée Puis installez cuDNN

#Installation de cuDNN
$ wget https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
$ wget https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb

Redémarrez ici Ceci termine l'installation de CUDA et cuDNN.

Installez le cmake requis à l'avance

$ sudo apt -y install cmake

Télécharger dlib

Vérifiez dlib official et téléchargez le dernier code source

$ wget http://dlib.net/files/dlib-19.20.tar.bz2
$ tar -jxvf dlib-19.20.tar.bz2

installation de dlib

Installer à l'aide de setup.py dans le dossier dlib-19.20 dlib-19.20 n'a pas besoin d'être installé à l'aide de la commande make De plus, il existe de nombreux articles dont vous avez besoin pour ajouter l'option -yes USE_AVX_INSTRUCTIONS --yes DLIB_USE_CUDA```. Dans la version 19.20, l'option yes est supprimée et activée par défaut, il n'est donc plus nécessaire d'ajouter l'option.

$ cd dlib-19.20
$ python3 setup.py install

Assurez-vous que CUDA et cuDNN sont activés lorsque setup.py est exécuté. Si le message suivant ne s'affiche pas, CUDA et cuDNN seront installés avec l'état désactivé.

-- Found CUDA: /usr/local/cuda-10.2 (found suitable version "10.2", minimum required is "7.5") 
-- Looking for cuDNN install...
-- Found cuDNN: /usr/lib/x86_64-linux-gnu/libcudnn.so
-- Building a CUDA test project to see if your compiler is compatible with CUDA...
-- Checking if you have the right version of cuDNN installed.
-- Enabling CUDA support for dlib.  DLIB WILL USE CUDA

Je reçois également un avertissement `` Aucune bibliothèque BLAS trouvée, donc en utilisant le BLAS intégré de dlib '', qui peut être ignoré (peut-être)

Vérification

Démarrez python en mode interactif et si DLIB_USE_CUDA vaut True, il se termine avec succès. Si c'est False ici, CUDA et cuDNN ne sont pas appliqués et sont installés.

$ python3
Python 3.7.6 (default, Jan  8 2020, 19:59:22) 
Type "help", "copyright", "credits" or "license" for more information.
>>> import dlib
>>> dlib.DLIB_USE_CUDA
True

Recommended Posts

Comment installer DLIB avec 2020 / CUDA activé
Comment installer python-pip avec ubuntu20.04LTS
Comment installer mysql-connector avec pip3
Comment installer Anaconda avec pyenv
Comment installer zsh (avec la personnalisation .zshrc)
Comment installer python3 avec docker centos
Comment installer Python
Comment installer pip
Comment installer Archlinux
Comment installer python
Comment installer BayesOpt
Comment installer Nbextensions
Comment installer Prover9
Comment installer OpenGM sur OSX avec macports
[2020.8 dernière] Comment installer Python
Comment installer Python [Windows]
Comment installer Tabpy 1.0 (version 2020-01)
Comment installer Caffe sur OS X avec macports
Comment mettre à jour avec SQLAlchemy?
Comment lancer avec Theano
Comment installer mkl numpy
Comment modifier avec SQLAlchemy?
Comment séparer les chaînes avec ','
Comment installer le blog Pelican
Comment faire RDP sur Fedora31
Comment supprimer avec SQLAlchemy?
Comment installer NPI + envoyer un message à la ligne avec python
Comment installer Python2.7 python3.5 avec pyenv (sur RHEL5 CentOS5) (novembre 2016)
Comment installer Theano sur Mac OS X avec homebrew
Comment installer Python à l'aide d'Anaconda
Python: comment utiliser async avec
Comment installer mysql-connector-python sur Mac
Comment installer et utiliser Tesseract-OCR
Que faire avec l'installation de Magics
Pour utiliser virtualenv avec PowerShell
Comment installer Graph-Tool sur macOS
Comment gérer les données déséquilibrées
Comment installer VMware-Tools sur Linux
Comment installer pycrypto sur Windows
Comment installer OpenCV sur Mac
Comment installer MBDyn (Linux Ubuntu)
Comment installer PyPy sur CentOS
Comment démarrer avec Scrapy
Installez facilement pyspark avec conda
Comment démarrer avec Python
Comment installer TensorFlow sur CentOS 7
Comment gérer l'erreur DistributionNotFound
Comment installer et configurer Blackbird
Comment démarrer avec Django
Comment installer et utiliser Graphviz
Comment augmenter les données avec PyTorch
Comment calculer la date avec python
Comment installer Maven sur CentOS
Comment INNER JOIN avec SQL Alchemy
Comment installer Music 21 sur Windows