Comment vérifier la version de CUDA sur Ubuntu (Guide de compatibilité cuDNN inclus)

目次

1. Introduction

CUDA (Compute Unified Device Architecture) est une plateforme de calcul parallèle développée par NVIDIA qui utilise les GPU. Elle est largement utilisée pour l’apprentissage automatique, le deep learning, le rendu 3D et de nombreuses autres charges de travail computationnelles.

Lorsque vous utilisez CUDA sous Ubuntu, il est important de vérifier votre version de CUDA pour les raisons suivantes :

Compatibilité du pilote

CUDA ne prend en charge que certaines versions de pilotes NVIDIA, et une incompatibilité peut empêcher le bon fonctionnement de CUDA.

Compatibilité des bibliothèques

Des bibliothèques comme TensorFlow et PyTorch nécessitent des versions spécifiques de CUDA et de cuDNN, il est donc essentiel de vérifier que les bonnes versions sont installées.

Éviter les conflits d’environnement

Si plusieurs versions de CUDA sont installées sur le système, vous devez savoir quelle version est actuellement active et pouvoir basculer entre elles selon les besoins.

Ce guide explique clairement comment vérifier la version de CUDA sous Ubuntu.

2. Comment vérifier la version de CUDA sous Ubuntu

Sous Ubuntu, vous pouvez vérifier votre version de CUDA en utilisant les méthodes suivantes.

Méthode 1 : Vérifier avec nvidia-smi (méthode la plus simple)

Le pilote NVIDIA inclut un outil appelé nvidia-smi (NVIDIA System Management Interface) qui affiche les informations du GPU.

Commande

nvidia-smi

Exemple de sortie

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 530.41.03    Driver Version: 530.41.03    CUDA Version: 12.1     |
+-----------------------------------------------------------------------------+

Points clés

  • La ligne CUDA Version: 12.1 indique la version maximale de CUDA prise en charge par le pilote.
  • Cela peut différer de la version réelle du Toolkit CUDA installé, il faut donc vérifier les méthodes suivantes également.

Méthode 2 : Vérifier avec nvcc -V (pour les développeurs)

Si CUDA est correctement installé, vous pouvez vérifier la version de nvcc, le compilateur CUDA.

Commande

nvcc -V

Exemple de sortie

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Sun_Jul_30_19:09:40_PDT_2023
Cuda compilation tools, release 12.1, V12.1.105

Points clés

  • release 12.1, V12.1.105 → Cela indique la version du Toolkit CUDA installé.
  • Elle peut ne pas correspondre à la version affichée par nvidia-smi.

Méthode 3 : Vérifier version.txt (vérification manuelle)

Si CUDA est installé sous /usr/local/cuda, le fichier version.txt contient les informations de version.

Commande

cat /usr/local/cuda/version.txt

Exemple de sortie

CUDA Version 12.1.105

Points clés

  • Utile lorsque nvcc -V n’est pas disponible.
  • Vous devez vous assurer que /usr/local/cuda est correctement lié.

3. Comment vérifier la version de cuDNN

cuDNN (CUDA Deep Neural Network) est une bibliothèque d’apprentissage profond utilisée avec CUDA.
Il est important de vérifier la version de cuDNN en même temps que celle de CUDA.

Méthode 1 : Vérifier cudnn_version.h

La version de cuDNN est stockée dans le fichier d’en-tête cudnn_version.h.

Commande

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

Exemple de sortie

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 9
#define CUDNN_PATCHLEVEL 1

Points clés

  • Indique que cuDNN 8.9.1 est installé.
  • La commande grep permet d’extraire facilement l’information de version.
  • cuDNN et CUDA doivent être compatibles ; vérifiez l’appariement correct.

Méthode 2 : Vérifier avec dpkg (uniquement pour les distributions basées sur Debian)

Sur les systèmes basés sur Debian comme Ubuntu, vous pouvez vérifier les paquets cuDNN installés avec dpkg.

Commande

dpkg -l | grep libcudnn

Exemple de sortie

ii  libcudnn8    8.9.1-1+cuda12.1    amd64    NVIDIA cuDNN Library

Points clés

  • La version de cuDNN (8.9.1) apparaît dans le nom du paquet.
  • cuda12.1 indique la version de CUDA compatible.

Utilisez ces méthodes pour vous assurer que votre environnement CUDA est correctement configuré.

4. Comment gérer plusieurs versions de CUDA installées

Sous Ubuntu, il est possible d’installer plusieurs versions de CUDA. Cependant, cela peut entraîner une confusion quant à la version actuellement active.
Dans ces cas‑de‑figure, vous devez basculer manuellement vers la version correcte.

Méthode 1 : basculer de version avec update-alternatives

Ubuntu vous permet de changer de version de CUDA à l’aide de l’outil update-alternatives.

Vérifier les paramètres actuels

update-alternatives --query cuda

Basculement de version

sudo update-alternatives --config cuda

Exemple de sortie

There are 3 choices for the alternative cuda (providing /usr/local/cuda).

  Selection    Path                Priority   Status
------------------------------------------------------------
* 0            /usr/local/cuda-11.8  100       auto mode
  1            /usr/local/cuda-10.2  50        manual mode
  2            /usr/local/cuda-11.8  100       manual mode
  3            /usr/local/cuda-12.1  110       manual mode

Press <enter> to keep the current choice[*], or type selection number:

Points clés

  • L’exécution de update-alternatives --config cuda affiche toutes les versions de CUDA disponibles.
  • Vous pouvez choisir la version à utiliser en saisissant le numéro correspondant.
  • Les modes auto et manual sont disponibles ; choisissez le mode manual pour un basculement manuel.

Méthode 2 : configurer manuellement les liens symboliques

Vous pouvez également activer une version spécifique de CUDA en ajustant manuellement les liens symboliques.

Vérifier le lien symbolique existant

ls -l /usr/local/cuda

Exemple de sortie

lrwxrwxrwx 1 root root 20 Feb  1 12:34 /usr/local/cuda -> /usr/local/cuda-11.8

Modifier la version de CUDA

sudo rm /usr/local/cuda
sudo ln -s /usr/local/cuda-12.1 /usr/local/cuda

Vérifier

ls -l /usr/local/cuda

Points clés

  • /usr/local/cuda est utilisé comme chemin CUDA par défaut, donc mettre à jour ce lien change la version active.
  • L’utilisation de ln -s rend le basculement entre les versions simple.

En utilisant ces méthodes, vous pouvez gérer plusieurs installations de CUDA et vous assurer que la bonne version est active.

5. Questions fréquentes (FAQ)

Cette section résume les problèmes courants liés à la vérification des versions de CUDA. Utilisez‑la pour le dépannage.

Q1 : nvcc -V introuvable !

Si la commande nvcc est absente, le chemin CUDA n’est peut‑être pas configuré.

Solution 1 : vérifier si CUDA est installé

ls /usr/local/cuda/

Solution 2 : ajouter nvcc au PATH

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

Après cela, exécutez à nouveau nvcc -V pour confirmer que cela fonctionne.

Q2 : pourquoi la version de CUDA affichée par nvidia-smi est‑elle différente ?

La version de CUDA affichée par nvidia-smi représente la version maximale de CUDA prise en charge par le pilote NVIDIA.

Exemple de vérification :

nvidia-smi

Exemple de sortie :

CUDA Version: 12.1

Cependant, cela n’indique pas la version du Toolkit CUDA installé. Utilisez nvcc -V ou version.txt pour vérifier la version réellement installée.

Q3 : comment vérifier la compatibilité entre CUDA et cuDNN ?

Le moyen le plus fiable est de consulter la matrice de support officielle de NVIDIA.

Site officiel :

NVIDIA cuDNN Support Matrix

Vous devez également confirmer la compatibilité en vérifiant les versions installées :

Vérifier la version de CUDA

nvcc -V

Vérifier la version de cuDNN

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

En gérant correctement ces versions, vous éviterez les problèmes liés à CUDA et cuDNN.

6. Résumé

Cet article a expliqué comment vérifier la version de CUDA dans un environnement Ubuntu.
Passons en revue les points essentiels.

Comment vérifier les versions de CUDA

MethodCommandDescription
nvidia-sminvidia-smiShows the CUDA version supported by the driver
nvcc -Vnvcc -VShows the installed CUDA Toolkit version
version.txtcat /usr/local/cuda/version.txtManually check the installed CUDA version

Comment vérifier cuDNN

MethodCommandDescription
cudnn_version.hcat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2Check version from the header file
dpkgdpkg -l | grep libcudnnCheck installed cuDNN packages

Comment basculer de version de CUDA

MethodCommandDescription
update-alternativessudo update-alternatives --config cudaSwitch between multiple CUDA versions
Symbolic Linksudo ln -s /usr/local/cuda-XX.X /usr/local/cudaManually switch CUDA versions

Points à retenir

  • Vérifiez toujours votre version de CUDA avant le développement
  • Vérifiez la compatibilité entre CUDA et cuDNN
  • Comprenez comment changer de version lorsqu’il existe plusieurs installations de CUDA

En gérant correctement votre environnement, vous pouvez maximiser les performances et la stabilité de CUDA.
Nous espérons que ce guide vous aidera à vérifier efficacement les versions de CUDA sur Ubuntu.

年収訴求