Comment vérifier la version de CUDA sur Ubuntu [Guide de commande facile]

目次

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, l’apprentissage profond, le rendu 3D et diverses autres tâches de calcul.

Lors de l’utilisation de CUDA dans un environnement Ubuntu, il est crucial de vérifier la version de CUDA pour les raisons suivantes :

Compatibilité du pilote

CUDA nécessite une version spécifique du pilote NVIDIA pour fonctionner correctement. Si les versions sont incompatibles, CUDA peut ne pas fonctionner correctement.

Compatibilité des bibliothèques

Des bibliothèques comme TensorFlow et PyTorch requièrent des versions spécifiques de CUDA et cuDNN. Il est essentiel de s’assurer que vous avez installé la version appropriée.

Éviter la confusion du système

Si plusieurs versions de CUDA sont installées sur le système, il est nécessaire d’identifier quelle version est active et de basculer entre les versions selon les besoins.

Dans cet article, nous fournirons une explication claire de comment vérifier la version de CUDA sur Ubuntu.

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

Dans un environnement Ubuntu, vous pouvez vérifier la version de CUDA en utilisant les méthodes suivantes :

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

Le pilote NVIDIA comprend un outil appelé nvidia-smi (NVIDIA System Management Interface) qui vous permet de vérifier l’état de votre GPU.

Commande d’exécution

nvidia-smi

Exemple de sortie

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

Points clés

  • Le CUDA Version: 12.1 affiché ici représente la version maximale de CUDA prise en charge par le pilote NVIDIA.
  • Cela ne correspond pas toujours à la version du kit d’outils CUDA installée, il est donc recommandé de vérifier à l’aide de méthodes supplémentaires.

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

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

Commande d’exécution

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

  • La partie qui indique release 12.1, V12.1.105 indique la version du kit d’outils CUDA installée.
  • Cela peut différer de la version affichée par nvidia-smi, soyez donc prudent.

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

Si CUDA est installé dans /usr/local/cuda, les informations de version sont enregistrées dans le fichier version.txt.

Commande d’exécution

cat /usr/local/cuda/version.txt

Exemple de sortie

CUDA Version 12.1.105

Points clés

  • Cette méthode est utile si la commande nvcc -V n’est pas disponible.
  • Assurez-vous que /usr/local/cuda est correctement lié à la version CUDA souhaitée.

3. Comment vérifier la version de cuDNN

cuDNN (CUDA Deep Neural Network) est une bibliothèque conçue pour l’apprentissage profond et est utilisée en combinaison avec CUDA.
En plus de vérifier la version de CUDA, il est également important de vérifier la version de cuDNN.

Méthode 1 : vérifier le fichier cudnn_version.h

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

Commande d’exécution

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

  • Cette sortie confirme que cuDNN 8.9.1 est installé.
  • Utiliser la commande grep vous permet de récupérer facilement les informations de version de cuDNN.
  • Étant donné que cuDNN doit être compatible avec CUDA, il est important de vérifier la combinaison correcte des versions.

Méthode 2 : vérifier avec la commande dpkg (pour Linux basé sur Debian)

Sur Ubuntu et d’autres distributions Linux basées sur Debian, vous pouvez vérifier la version installée de cuDNN en utilisant la commande dpkg.

Commande d’exécution

dpkg -l | grep libcudnn

Exemple de sortie

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

Points clés

  • La partie libcudnn8 8.9.1-1+cuda12.1 confirme la version cuDNN (8.9.1) installée.
  • La partie cuda12.1 indique la version CUDA compatible (12.1).

En utilisant ces méthodes, vous pouvez vous assurer que votre environnement CUDA est correctement configuré.

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

Dans un environnement Ubuntu, plusieurs versions de CUDA peuvent être installées. Cependant, cela peut parfois entraîner de la confusion quant à la version actuellement active.
Dans de tels cas, vous devez passer à la version appropriée.

Méthode 1 : basculer en utilisant update-alternatives

Sur Ubuntu, vous pouvez utiliser update-alternatives pour changer de version CUDA.

Vérifier les paramètres actuels

update-alternatives --query cuda

Basculer la version CUDA

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 affichera une liste des versions CUDA disponibles.
  • Vous pouvez sélectionner la version CUDA souhaitée en entrant le numéro correspondant.
  • auto mode et manual mode sont disponibles ; choisissez manual mode si vous souhaitez changer de version manuellement.

Méthode 2 : définir manuellement un lien symbolique

Vous pouvez également changer de version CUDA en modifiant le lien symbolique.

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

Changer la version CUDA

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

Vérifier le changement

ls -l /usr/local/cuda

Points clés

  • /usr/local/cuda sert de chemin CUDA par défaut. Modifier ce lien change la version CUDA.
  • En utilisant la commande ln -s, vous pouvez facilement changer la version CUDA sans modifier les configurations système.

Avec ces méthodes, vous pouvez gérer efficacement plusieurs versions de CUDA et vous assurer d’utiliser la version correcte pour vos besoins.

5. Questions fréquemment posées (FAQ)

Voici quelques questions fréquentes concernant la vérification de la version de CUDA. Si vous rencontrez des problèmes, consultez ces solutions.

Q1 : nvcc -V Commande introuvable !

Si la commande nvcc n’est pas trouvée, il se peut que CUDA ne soit pas installé correctement ou que son chemin ne soit pas défini.

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 avoir exécuté ces commandes, essayez de lancer nvcc -V à nouveau pour vérifier si la version s’affiche correctement.

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

La version CUDA affichée par nvidia-smi représente la version maximale de CUDA prise en charge par le pilote NVIDIA, et ne correspond pas forcément à la version du kit d’outils CUDA installée.

Comment vérifier :

nvidia-smi

Exemple de sortie :

CUDA Version: 12.1

Pour vérifier la version réelle de CUDA installée, utilisez nvcc -V ou consultez le fichier version.txt.

Q3 : Comment la compatibilité entre CUDA et cuDNN ?

Le meilleur moyen de vérifier la compatibilité entre CUDA et cuDNN est de se référer à la matrice de support officielle de NVIDIA.

Documentation officielle :

Matrice de support cuDNN de NVIDIA

De plus, vous pouvez vérifier les versions installées à l’aide des commandes suivantes :

Vérifier la version CUDA

nvcc -V

Vérifier la version cuDNN

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

En gérant correctement votre environnement, vous pouvez éviter les problèmes de compatibilité entre CUDA et cuDNN.

6. Résumé

Dans cet article, nous avons expliqué comment vérifier la version de CUDA dans un environnement Ubuntu.
Passons en revue les points clés.

Méthodes pour vérifier la version de CUDA

MéthodeCommandDescription
nvidia-sminvidia-smi

Affiche la version CUDA prise en charge par le pilote NVIDIA

nvcc -Vnvcc -VAffiche la version réellement installée du kit d’outils CUDA
version.txtcat /usr/local/cuda/version.txtVérifiez manuellement la version CUDA

Méthodes pour vérifier la version de cuDNN

MéthodeCommandDescription
cudnn_version.hcat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2Vérifiez la version dans le fichier d’en-tête
dpkgdpkg -l | grep libcudnnVérifiez la version cuDNN installée

Comment changer de version de CUDA

MéthodeCommandDescription
update-alternativessudo update-alternatives --config cudaBasculer entre plusieurs versions CUDA
Lien symboliquesudo ln -s /usr/local/cuda-XX.X /usr/local/cudaModifier manuellement la version CUDA

Points clés à retenir

  • Il est important d’identifier correctement la version de CUDA
  • Assurez la compatibilité entre CUDA et cuDNN
  • Si vous utilisez plusieurs versions de CUDA, comprenez comment basculer entre elles

En gérant correctement votre environnement, vous pouvez maximiser les avantages de CUDA.
Nous espérons que cet article vous aidera à vérifier la version de CUDA dans votre environnement Ubuntu.

Articles associés

年収訴求