Comment installer CUDA sur Ubuntu 20.04

CUDA est une plate-forme informatique parallèle et un modèle de programmation qui offre une expérience utilisateur remarquable lors de l'utilisation du GPU pour l'informatique à usage général au quotidien. CUDA signifie C calculer U nifié D appareil A architecture. Il est créé par NVIDIA.

CUDA comprend la boîte à outils CUDA (compilateur, profil et débogueur), le pilote logiciel et le SDK CUDA. La couche logicielle donne un accès direct au jeu d'instructions virtuelles du GPU et aux éléments de calcul parallèles. Pour les recherches d'apprentissage en profondeur et les développeurs de framework, utilisez cuDNN pour une accélération GPU haute performance. C'est une bibliothèque de primitives pour les réseaux de neurones profonds.

Dans ce guide, nous apprenons comment installer CUDA et CuDNN sur Ubuntu 20.04 . Il y a deux façons de procéder. Vous pouvez installer CUDA à partir du référentiel Ubuntu - qui est l'approche la plus simple - ou installer à partir du référentiel CUDA qui fournit la dernière version de CUDA.

Prérequis :

Au moins 20 Go d'espace disque disponible
2 Go de RAM recommandés
Le pilote graphique NVIDIA doit être installé. Le pilote fourni avec CUDA n'est peut-être pas la dernière version.

Méthode 1 :Installer CUDA à partir du référentiel Ubuntu

Commencez par mettre à jour les listes de packages sur votre instance Ubuntu 20.04.

$ sudo apt update

Ensuite, installez la boîte à outils CUDA à l'aide du gestionnaire de packages APT comme suit. Notez que l'installation est intensive, et à ce titre assurez-vous d'avoir une connexion internet rapide et stable et un minimum de 10G d'espace disque.

$ sudo apt install nvidia-cuda-toolkit

Une fois l'installation terminée, nous devons ajouter CUDA à PATH afin d'informer le shell de l'emplacement de CUDA. Pour ce faire, nous allons spécifier le PATH dans le fichier .bashrc. Ouvrez donc le fichier à l'aide de votre éditeur de texte en ligne de commande.

$ sudo vim ~/.bashrc

À la toute fin du fichier, copiez et collez les lignes de code suivantes.

# 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

Ensuite, rechargez les modifications dans le fichier .bashrc.

$ source ~/.bashrc

Maintenant, vérifiez et confirmez la version de CUDA installée à l'aide de la commande suivante.

$ nvcc --version

Méthode 2 :Installer CUDA à partir du référentiel CUDA

Comme mentionné précédemment, le référentiel CUDA fournit la dernière version de CUDA. Donc, si vous souhaitez installer la dernière version, suivez les instructions décrites ici.

Tout d'abord, téléchargez la broche du référentiel CUDA :

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin

Ensuite, déplacez l'épingle vers /etc/preferences.d répertoire et renommez-le en cuda-repository-pin-600 .

$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600

Ensuite, téléchargez le fichier CUDA Debian. Sachez que le fichier est assez volumineux - 2,3 Go pour être précis au moment de la rédaction de ce guide. Assurez-vous d'avoir une bonne connexion Internet et suffisamment d'espace disque.

$ wget https://developer.download.nvidia.com/compute/cuda/11.4.2/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

Une fois téléchargé, utilisez le dpkg utilitaire pour exécuter le fichier.

sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.2-470.57.02-1_amd64.deb

Ensuite, installez la clé CUDA GPG comme suit :

$ sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub

Enfin, mettez à jour les listes de packages et installez CUDA à l'aide du gestionnaire de packages APT.

$ sudo apt update

$ sudo apt install cuda -y

Installer CuDNN

CuDNN n'est pas fourni avec Cuda. Pour télécharger CuDNN, vous devez vous inscrire pour devenir membre du programme de développement NVIDIA qui est gratuit.

Après vous être inscrit au programme pour développeurs NVIDIA, vous pouvez télécharger la dernière version de cuDNN. Ici, je télécharge la bibliothèque cuDNN pour Linux.

Tout d'abord, extrayez le package cuDNN :

$ tar -xzvf cudnn-10.2-linux-x64-v8.2.4.15.tgz

Copiez ensuite les fichiers suivants dans le répertoire CUDA :

$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

Conclusion

Dans ce tutoriel, nous avons appris à installer CUDA et cuDNN sur Ubuntu 20.04. Merci d'avoir lu, veuillez laisser vos commentaires et suggestions.