Comment installer Pytorch sur Ubuntu?

Mis à jour le Sept. 12, 2022, 1:07 p.m.

1-Installation de Pytorch sur CPU
2-Vérification de l'installation de Pytorch sur CPU
3-Installation de Pytorch sur GPU
4-Vérification de l'installation de Pytorch sur GPU

Installation de Pytorch sur CPU:

Aller sur https://pytorch.org/

Scroller jusqu'à cette image et sélectionner les éléments pertinents.

Si vous n'avez pas de GPU:

Maintenant il ne vous reste qu'à copier coller la commande qui apparait dans votre terminal, sous votre environnement virtuel si vous en avez un, ce que je vous conseille. Si vous voulez en créer un, rendez-vous ici: comment créer un environnement virtuel sous ubuntu

Vérification de l'installation de Pytorch sur CPU:

Vérifiez votre installation en tapant dans votre terminal:

python

>>> import torch

Super cela fonctionne!!!

Installation de Pytorch sur GPU:

Si vous avez un GPU:

Pour ma part j'ai un pc sous linux, je peux installer pip, pour Python et j'ai cuda 11.4.

On peut vérifier sa version de cuda tapant la commande suivante dans le terminal:

nvidia-smi

Resultat:

OK, j'ai la version 11.4 de CUDA.

Maintenant je sélectionnne ce qui correspond, mais il y a un problème, pas de cuda 11.4

La 11.4 n'apparait pas, je remplace cu113 par cu114 dans la commande. Cela fonctionnera pour la 11.4 mais risque de ne pas fonctionner pour les versions supérieures à 11.6

Ma commande d'install donnera ça:

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu114

Vérification de l'installation de Pytorch sur GPU:

Vérifiez que votre installation est bonne, tapez dans votre terminal:

python

>>> import torch

>>> torch.cuda.is_available()

Super, vous pouvez maintenant entrainer votre GPU sur des réseaux de neurones

Voilà pour votre installation.