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
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.
$ sudo apt -y install cmake
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
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)
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