Installer et configurer PyTorch

  • Article

Dans l’étape précédente de ce tutoriel, nous avons abordé les notions de base de PyTorch et les conditions préalables à l’utilisation de ce dernier pour créer un modèle d’apprentissage automatique. Ici, nous allons l’installer sur votre ordinateur.

Télécharger PyTorch

Tout d’abord, vous devez configurer un environnement Python.

Nous vous recommandons de configurer un environnement Python virtuel dans Windows, en utilisant Anaconda comme gestionnaire de package. La suite de ce programme d’installation suppose que vous utilisez un environnement Anaconda.

  1. Téléchargez et installez Anaconda ici. Sélectionnez Anaconda 64-bit installer for Windows Python 3.8.

Important

N’oubliez pas d’installer Python 3.x. Actuellement, PyTorch sur Windows prend en charge uniquement Python 3.x. Python 2.x n’est pas pris en charge.

Anaconda installers

Une fois l’installation terminée, vérifiez vos versions Anaconda et Python.

  1. Ouvrez Anaconda Manager via Démarrer - Anaconda3 - Anaconda PowerShell Prompt et testez vos versions :

Vous pouvez vérifier votre version de Python en exécutant la commande suivante : python –-version

Vous pouvez vérifier votre version d’Anaconda en exécutant la commande suivante : conda –-version

Anaconda command line prompt

À présent, vous pouvez installer le package PyTorch à partir de fichiers binaires via Conda.

  1. Accédez à https://pytorch.org/.

Sélectionnez les informations d’installation PyTorch appropriées :

  • Build de PyTorch – stable.
  • Votre système d’exploitation – Windows
  • Package - Conda
  • Langue – Python
  • Plateforme de calcul – processeur ou choisissez votre version de Cuda. Dans ce didacticiel, vous allez former et inférencer le modèle sur le processeur, mais vous pouvez également utiliser un GPU Nvidia.
  1. Ouvrez Anaconda Manager et exécutez la commande telle qu’elle est spécifiée dans les instructions d’installation.
conda install pytorch torchvision torchaudio cpuonly -c pytorch

Anaconda installation

  1. Confirmez et terminez l’extraction des packages requis.

Confirm Anaconda installation

Vérifions l’installation de PyTorch en exécutant l’exemple de code PyTorch pour construire un tenseur initialisé de manière aléatoire.

  1. Ouvrez l’invite PowerShell Anaconda et exécutez la commande suivante.
python

Entrez ensuite la commande suivante :

import torch 

x = torch.rand(2, 3) 

print(x)

La sortie doit être un tenseur 5x3 aléatoire. Bien que différents, les nombres doivent ressembler à ce qui suit.

Verify Pytorch functionality

Remarque

Vous voulez en savoir plus ? Visitez le site web officiel de PyTorch

Étapes suivantes

Maintenant que nous avons installé PyTorch, nous sommes prêts à configurer les données pour notre modèle.