PyTorch est l’un des cadres d’apprentissage automatique open-source les plus populaires, développé par le laboratoire de recherche en IA de Facebook. Il est largement utilisé pour les applications d’apprentissage profond, y compris la vision par ordinateur et le traitement du langage naturel. Si vous utilisez AlmaLinux, une distribution Linux stable et fiable basée sur RHEL, ce guide vous montrera comment installer PyTorch rapidement et efficacement.

Conditions préalables

Avant de commencer, assurez-vous que

  • Vous disposez d’une instance d’AlmaLinux 8 ou 9 en cours d’exécution

  • Vous avez les privilèges root ou sudo

  • Python 3.7 ou supérieur est installé

Etape 1 : Mettre à jour votre système

Ouvrez votre terminal et exécutez la commande suivante pour mettre à jour tous les paquets du système :

sudo dnf update -y

En option, installez EPEL pour accéder à des paquets supplémentaires :

sudo dnf install epel-release -y

Étape 2 : Installer Python et pip

Vérifiez si Python est déjà installé :

python3 --version

S’il n’est pas installé, vous pouvez l’installer avec :

sudo dnf install python3 python3-pip -y

Étape 3 : Créer un environnement virtuel (optionnel)

Pour garder votre environnement Python propre et isolé, c’est une bonne idée de créer un environnement virtuel :

python3 -m venv pytorch_env
source pytorch_env/bin/activate

Étape 4 : Installer PyTorch à l’aide de pip

Pour installer PyTorch avec le support CPU uniquement :

pip install torch torchvision torchaudio

Si vous souhaitez bénéficier de l’accélération GPU et que vous disposez d’un GPU NVIDIA pris en charge avec les pilotes appropriés et CUDA installé, utilisez la commande appropriée. Par exemple, pour installer PyTorch avec le support de CUDA 11.8 :

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Vous trouverez les dernières options d’installation sur le site officiel de PyTorch :
https://pytorch.org/get-started/locally

Étape 5 : Vérifier l’installation

Pour vérifier que PyTorch a été correctement installé, exécutez :

python

Puis entrez ce qui suit dans l’interpréteur de commandes Python :

import torch
print(torch.__version__)
print("CUDA available:", torch.cuda.is_available())

Si tout fonctionne, vous verrez la version installée de PyTorch et si CUDA est disponible.

Conseils de dépannage

  • Si vous rencontrez des problèmes avec pip ou les certificats SSL, mettez à jour pip :

pip install --upgrade pip

  • Si vous utilisez CUDA, assurez-vous que les pilotes NVIDIA et la boîte à outils CUDA sont correctement installés et compatibles avec la version de PyTorch que vous installez.

Conclusion

Vous avez réussi à installer PyTorch sur AlmaLinux. Que vous développiez des modèles d’apprentissage profond ou que vous fassiez des expériences, PyTorch fournit un cadre flexible et puissant pour l’apprentissage automatique sur les systèmes Linux. Pour des configurations plus avancées, envisagez l’intégration avec Jupyter Notebook ou l’utilisation de conteneurs Docker.