Maison  >  Article  >  Opération et maintenance  >  Comment installer PyTorch sous Linux

Comment installer PyTorch sous Linux

PHPz
PHPzavant
2023-05-14 14:25:134769parcourir

1. Introduction à PyTorch

PyTorch est une bibliothèque d'apprentissage automatique Python open source, basée sur Torch, utilisée pour le traitement du langage naturel et d'autres applications. En janvier 2017, PyTorch a été lancé sur la base de Torch par le Facebook Artificial Intelligence Research Institute (FAIR). Le prédécesseur de PyTorch est Torch. Sa couche sous-jacente est la même que celle du framework Torch, mais une grande partie du contenu a été réécrite en Python. Elle est non seulement plus flexible, prend en charge les graphiques dynamiques, mais fournit également une interface Python. Développé par l'équipe Torch7, il s'agit d'un premier framework d'apprentissage profond Python qui permet non seulement une puissante accélération GPU, mais prend également en charge les réseaux de neurones dynamiques. PyTorch peut être considéré comme numpy avec prise en charge GPU, et il peut également être considéré comme un puissant réseau neuronal profond avec fonction de dérivation automatique. Outre Facebook, il a été adopté par des institutions telles que Twitter, CMU et Salesforce.

2. Étapes d'installation

1. Sélection du système d'exploitation

   Consultez le site officiel de PyTorch, vous pouvez voir que PyTorch prend en charge Linux, Mac, les plates-formes Windows, prend en charge conda, pip, le code source et d'autres méthodes d'installation, et prend également en charge les plates-formes informatiques CPU, cuda et ROCm. Lorsque nous cliquons sur la sélection de l'environnement, nous pouvons constater qu'actuellement, seul le système Linux prend en charge toutes les langues, toutes les méthodes d'installation et toutes les plates-formes informatiques. Nous choisissons donc le système d'exploitation Linux comme système d'exploitation. environnement système. De plus, les calculs d'apprentissage automatique nécessitent une version supérieure de la glibc, les versions du noyau et glbic de centos sont inférieures et les versions du noyau d'Ubuntu sont plus récentes, il est donc recommandé d'utiliser le système d'exploitation Ubuntu pour les hôtes d'apprentissage automatique. Actuellement, la version minimale d'Ubuntu prise en charge par la mise à jour cuda est 18.04, il est donc recommandé d'utiliser un système d'exploitation Ubuntu 18.04 ou supérieur.

Comment installer PyTorch sous Linux

wuhs@s169:~$ cat /etc/os-release
NAME="Ubuntu"
VERSION="18.04.6 LTS (Bionic Beaver)"

2, installation d'Anaconda3

 &emsp ; Comme indiqué ci-dessus, PyTorch prend en charge une variété de méthodes d'installation. Le blogueur prévoit d'utiliser la méthode d'installation conda. Nous pouvons d'abord créer différents environnements virtuels en fonction de nos besoins. environnement virtuel. L'environnement virtuel se soutient mutuellement. Pour l'installation d'anaconda dans l'environnement Ubuntu, consultez le billet de blog Ubuntu Anaconda3 Installation.

wuhs@s169 :~$ wget https://mirrors.bfsu.edu.cn/anaconda/archive/Anaconda3-2022.10-Linux-x86_64.sh
wuhs@s169 :~$ sh Anaconda3-2022.10-Linux-x86_64 .sh
wuhs@s169 :~$ source ~/.bashrc

3. Vérifiez la version de Python

  Différentes versions de PyTorch nécessitent différentes versions de Python, donc après avoir installé anaconda3, nous vérifions la version actuelle de Python, l'initialisation par défaut est la dernière version de Python correspondant à la version actuelle d'anaconda3. Bien entendu, nous pouvons également utiliser conda pour créer la version d'environnement Python requise. Nous vérifions les exigences de correspondance des versions de PyTorch, torchvision et Python chez torchvision.

Comment installer PyTorch sous Linux

(base) wuhs@s169:~$ python -V
Python 3.9.13

4. Installez PyTorch

  La deuxième étape est la suivante À partir du site officiel de PyTorch, nous. peut choisir le système d'exploitation. Après avoir déterminé la méthode d'installation, le langage de programmation et la plate-forme informatique, la commande d'installation correspondante est générée.

(base) wuhs@s169 :~$ conda install pytorch torchvision torchaudio cpuonly -c pytorch
Collecte des métadonnées du package (current_repodata.json) : terminé
Environnement de résolution : terminé

## Plan de package ##

Comment installer PyTorch sous Linux


Procéder ([o]/n) ? y

5. Vérification de la version

(base) wuhs@s169 :~$ python
Python 3.9.13 (principal, 25 août 2022, 23:26 : 10)
[GCC 11.2.0] :: Anaconda, Inc. sur Linux
Tapez « aide », « copyright », « crédits » ou « licence » pour plus d'informations.
>>> >> torch.
version‘1.13.1’
>>>

3. Installation de la version spécifiée

1. Créer un environnement virtuel

(base) wuhs@s169 : ~$ conda create -n pytorch python=3.9


(base) wuhs@s169:~$ conda activate pytorch
(pytorch) wuhs@s169:~$

2 Installez la version spécifiée de PyTorch

&emsp. ;  Lors de l'installation de la version spécifiée de PyTorch, nous devons vérifier la version correspondante dans le canal PyTorch du site officiel de GitHub. Lors de l'installation de conda, spécifiez le numéro de version spécifique, vous pouvez consulter le site officiel d'anaconda. . Pour la relation correspondante entre la version PyTorch et TorchAudio, voir

TorchAudio. Bien entendu, si nous spécifions la mauvaise version, une erreur sera signalée lors de l'installation. Nous pouvons vérifier quelle version du logiciel est erronée en fonction du message d'erreur, puis accéder au site officiel pour confirmer la correction et réinstaller.

Comment installer PyTorch sous Linux

Comment installer PyTorch sous Linux

(pytorch) wuhs@s169:~$ conda install pytorch2.12.0 torchvision=0.13.0 torchaudio0.12.0 cpuonly -c pytorch

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!

Déclaration:
Cet article est reproduit dans:. en cas de violation, veuillez contacter admin@php.cn Supprimer