Comment installer Pytorch sur Ubuntu?

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

Salut salut, et bienvenue dans le monde du deep learning.

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 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érifiez votre installation en tapant dans votre terminal:

python

>>> import torch

Super cela fonctionne!!!

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érifiez que votre installation est bonne, tapez dans votre terminal:

python

>>> import torch

>>> torch.cuda.is_available()

Voilà pour votre installation. Si vous n'avez pas configurez votre GPU aller ici: article à venir