web-dev-qa-db-fra.com

Comment installer les pilotes NVIDIA et CUDA dans Ubuntu?

J'ai un système avec une carte NVIDIA qui prend en charge les calculs de plus de 3,5 comparés à https://developer.nvidia.com/cuda-gpus . Comment installer CUDA et les pilotes NVIDIA dans Ubuntu sans télécharger les fichiers .deb de NVIDIA?

21
Terrance

Pilotes Ubuntu 18.04, CUDA 10.1, libcudnn 7.5.1 et NVIDIA 418.67:

NOTE 2019-06-10: Mises à jour récentes avec les versions CUDA 10.0 ou 10.1 ou NVIDIA 418.67 Le pilote, qui l’installe avec, n’a plus les bibliothèques 32 bits incluses, ce qui fera que Steam et la plupart des jeux ne fonctionneront plus. La version de libnvidia-gl-418:i386 installe uniquement la version 418.56 qui ne fonctionnera pas avec le pilote 418.67. Espérons que NVIDIA publiera bientôt une mise à jour pour cela. Voir: https://github.com/ValveSoftware/Steam-for-linux/issues/5778

REMARQUE: CUDA 9.x n'est pas disponible via le référentiel ubuntu1804 de NVIDIA. . J'ai cependant écrit une réponse pour CUDA 9.2 à https://askubuntu.com/a/1086993/231142

Ces instructions concernent l'installation de CUDA via le référentiel au lieu de l'installation .deb.

Les lignes suivantes que vous pouvez copier et coller dans une fenêtre de terminal. presse Ctrl+Alt+T ouvrir une fenêtre de terminal.

Supprimez tous les PPA CUDA pouvant être configurés, ainsi que le nvidia-cuda-toolkit s'il est installé:

Sudo rm /etc/apt/sources.list.d/cuda*
Sudo apt remove --autoremove nvidia-cuda-toolkit

Il est recommandé de supprimer également tous les pilotes NVIDIA avant d'installer de nouveaux pilotes:

Sudo apt remove --autoremove nvidia-*

Puis mettez à jour le système:

Sudo apt update

Récemment, je viens de découvrir que l'installation de CUDA fonctionne avec le graphics-drivers ppa. Si vous ne l'avez pas ajouté, ajoutez-le maintenant:

Sudo add-apt-repository ppa:graphics-drivers/ppa

Installez la clé:

Sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

Ajouter les pensions:

Sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

Sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda_learn.list'

Mettez à jour le système à nouveau:

Sudo apt update

Installez CUDA 10.1:

Sudo apt install cuda-10-1

Il devrait installer les pilotes NVIDIA 418.40 avec, car c’est ce qui est indiqué dans le référentiel. Voir: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/

Installez libcudnn7 7.5.1:

Sudo apt install libcudnn7

Ajoutez les lignes suivantes à votre fichier ~/.profile pour CUDA 10.1

# set PATH for cuda 10.1 installation
if [ -d "/usr/local/cuda-10.1/bin/" ]; then
    export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

Redémarrez l'ordinateur et vérifiez vos paramètres à la fin du redémarrage:

Vérifiez le compilateur NVIDIA Cuda avec nvcc --version:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Apr_24_19:10:27_PDT_2019
Cuda compilation tools, release 10.1, V10.1.168

Vérifiez la version de libcudnn /sbin/ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn:

terrance@terrance-ubuntu:~$ /sbin/ldconfig -N -v $(sed 's/:/ /' <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn
    libcudnn.so.7 -> libcudnn.so.7.5.1

Vérifiez le pilote NVIDIA avec nvidia-smi:

terrance@terrance-ubuntu:~$ nvidia-smi 
Sat Jun  1 09:38:07 2019       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.67       Driver Version: 418.67       CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 750 Ti  On   | 00000000:02:00.0  On |                  N/A |
| 40%   38C    P0     2W /  38W |    116MiB /  2000MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      2216      G   /usr/lib/xorg/Xorg                           110MiB |
|    0      2542      G   compton                                        1MiB |
+-----------------------------------------------------------------------------+
38
Terrance

Pilotes Ubuntu 16.04, CUDA 9.2 et NVIDIA 396:

REMARQUE: Le dépôt de NVIDIA ici a décidé de pousser les pilotes 410. Je vais faire des tests pour voir si je peux le configurer pour le pilote que vous voulez installer.

Ces instructions concernent l'installation de CUDA via le référentiel au lieu de l'installation .deb.

Les lignes suivantes que vous pouvez copier et coller dans une fenêtre de terminal. presse Ctrl+Alt+T ouvrir une fenêtre de terminal.

Supprimez tous les PPA CUDA pouvant être configurés, ainsi que le nvidia-cuda-toolkit s'il est installé:

Sudo rm /etc/apt/sources.list.d/cuda*
Sudo apt remove nvidia-cuda-toolkit

Il est recommandé de supprimer également tous les pilotes NVIDIA avant d'installer de nouveaux pilotes:

Sudo apt remove nvidia-*

Puis mettez à jour le système:

Sudo apt update

Installez la clé:

Sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub

Ajouter le repo:

Sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

Mise à jour pour le nouveau référentiel ajouté:

Sudo apt update

Installez CUDA 9.2:

Sudo apt install cuda-9-2

Il devrait installer les pilotes nvidia-396 avec, car c’est ce qui est indiqué dans le référentiel. Voir: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/

Vérifiez que CUDA 9.2 a été installé:

~$ ls /usr/local/cuda-9.2/
bin     include    libnvvp           nvml    samples  targets
doc     lib64      LICENSE           nvvm    share    tools
extras  libnsight  nsightee_plugins  README  src      version.txt

Maintenant, ajoutez ce qui suit à votre ~/.profile pour les PATH et LD_LIBRARY. Vous pouvez utiliser la commande gedit ~/.profile pour la modifier:

# set PATH for cuda 9.2 installation
if [ -d "/usr/local/cuda-9.2/bin/" ]; then
    export PATH=/usr/local/cuda-9.2/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-9.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

Redémarrez votre système.

Sudo reboot

Une fois le système installé, vous pouvez vérifier l’installation en tapant ce qui suit:

nvcc -V

Vous devriez voir quelque chose de similaire à ce qui suit:

~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

Et vous devriez voir les pilotes 396.xx installés:

~$ nvidia-smi
Thu May 17 07:38:54 2018        
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 396.44                 Driver Version: 396.44                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 760     Off  | 00000000:02:00.0 N/A |                  N/A |
| 49%   53C    P0    N/A /  N/A |    187MiB /  1999MiB |     N/A      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0                    Not Supported                                       |
+-----------------------------------------------------------------------------+

4
Terrance

Pilotes Ubuntu 14.04, CUDA 10.0 et NVIDIA 410:

Veuillez noter que je n'ai pas installé Ubuntu 14.04, donc je ne peux pas vérifier si ces étapes fonctionnent ou non. Mais selon http://developer.download.nvidia.com/compute/cuda/repos/ de NVIDIA, il devrait s'agir des mêmes étapes que celles énumérées dans les deux autres réponses.

Ces instructions concernent l'installation de CUDA via le référentiel au lieu de l'installation .deb.

Les lignes suivantes que vous pouvez copier et coller dans une fenêtre de terminal. presse Ctrl+Alt+T ouvrir une fenêtre de terminal.

Supprimez tous les PPA CUDA pouvant être configurés, ainsi que le nvidia-cuda-toolkit s'il est installé:

Sudo rm /etc/apt/sources.list.d/cuda*
Sudo apt remove nvidia-cuda-toolkit

Il est recommandé de supprimer également tous les pilotes NVIDIA avant d'installer de nouveaux pilotes:

Sudo apt remove nvidia-*

Puis mettez à jour le système:

Sudo apt update

Installez la clé:

Sudo apt-key adv --fetch-keys  http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/7fa2af80.pub

Ajouter le repo:

Sudo bash -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

Mettez à jour le système à nouveau:

Sudo apt update

Installez CUDA 10.0.

Sudo apt install cuda-10-0

Il devrait installer les pilotes nvidia-410 avec lui car ce sont ceux qui sont listés dans le repo. Voir: http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/

Ajoutez les lignes suivantes à votre fichier ~/.profile pour CUDA 10.0

# set PATH for cuda 10.0 installation
if [ -d "/usr/local/cuda-10.0/bin/" ]; then
    export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
    export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
fi

Redémarrez l'ordinateur et vérifiez vos paramètres à la fin du redémarrage:

Vérifiez le compilateur NVIDIA Cuda avec nvcc --version:

Vérifiez le pilote NVIDIA avec nvidia-smi:

2
Terrance