03/29/2024

Cum se instalează PyTorch pe Windows și Linux

În acest ghid, voi trece prin instalarea PyTorch pe Windows și Linux, folosind Anaconda pentru a gestiona instalarea pachetelor.

Instalarea se va face folosind instrumentul de linie de comandă conda încorporat în Anaconda.

PyTorch este o bibliotecă de învățare automată scrisă în Python și se bazează pe framework-ul Torch. A fost dezvoltat de Facebook și este comparabil cu Tensorflow de la Google. Este util în domeniile viziunii computerizate și procesării limbajului natural și a fost folosit de companii precum Tesla pentru a dezvolta software pentru pilot automat.

PyTorch este gratuit și open-source, licențiat sub BSD modificat și este sub Linux Foundation.

Cerințe preliminare

Pentru a urma acest tutorial, trebuie să aveți Anaconda instalat pe mașina la care lucrați.

Dacă nu îl aveți deja instalat, acest ghid despre cum să instalați Anaconda vă va ghida prin întregul proces. După ce ați urmat acest ghid, puteți continua la instalarea PyTorch.

Instalarea PyTorch pe Linux

Ca o practică bună, începeți prin a actualiza pachetele software din distribuția dvs. Linux. În cazul meu, folosesc Ubuntu și apt să-mi gestionez pachetele, așa că voi folosi următoarea comandă pentru a actualiza:

$ sudo apt update && apt upgrade 

Când ați terminat de actualizat pachetele, accesați pagina oficială de instalare a site-ului web PyTorch. Derulați în jos în pagină până când găsiți asistentul de instalare care arată ca cel de mai jos:

Folosind acest expert, veți putea să faceți clic pe diferitele opțiuni pentru a furniza informațiile și preferințele de sistem și, la rândul său, veți obține o comandă pe care o puteți utiliza în terminalul dvs. pentru a instala PyTorch.

  Cum să-ți clonezi instalarea Linux cu Clonezilla

După ce mi-am furnizat informațiile despre sistem, așa arată:

Am optat pentru versiunea Linux stabilă și voi folosi Conda pentru a-mi gestiona pachetele. De asemenea, am ales să folosesc PyTorch cu Python, mai degrabă decât C++/Java. Și îmi voi rula PyTorch pe un procesor, spre deosebire de un GPU.

În partea de jos a tabelului se află comanda pe care o pot folosi pentru a instala PyTorch, dar înainte de a rula această comandă, aș dori să creez un mediu virtual Anaconda numit pytorch.

Mediile virtuale vă permit să creați proiecte și să păstrați dependențele lor izolate de dependențele altor proiecte, prevenind astfel conflictele de dependențe. Unul dintre beneficiile Anaconda este că vă ajută să creați și să gestionați cu ușurință mediile virtuale.

Pentru a crea un mediu virtual în care versiunea Python este 3.7, voi introduce următoarea comandă:

conda create -n pytorch python=3.7

După ce mediul este creat, îl voi activa folosind următoarea comandă:

conda activate pytorch

Odată ce mediul este activ, voi rula comanda generată mai devreme pe site-ul web PyTorch pentru a instala PyTorch.

conda install pytorch torchvision torchaudio cpuonly -c pytorch

Urmați instrucțiunile pentru a instala PyTorch. Odată terminat, voi reporni sesiunea terminalului pentru a intra în vigoare.

Acum, pentru a verifica dacă PyTorch a fost instalat corect, vom încerca să-l importăm în shell-ul interactiv Python. Asigurați-vă că vă aflați în mediul virtual Pytorch folosind comanda:

conda activate pytorch

Odată ce vă aflați în mediul virtual Pytorch, deschideți shell-ul interactiv Python tastând comanda:

python

Odată ce începe sesiunea shell, scrieți următoarea linie de cod și apăsați ENTER

import torch

Dacă Python rulează fără erori, atunci instalarea a avut succes. Dar dacă ați primit o eroare Module Not Found, înseamnă că ceva a mers prost în timpul instalării. Puteți încerca să-l reinstalați din nou.

  Cum să scanați porturi deschise pe un computer Linux cu Zenmap

Instalarea PyTorch pe Windows

Pentru a începe, pe computerul dvs. Windows, căutați programul Anaconda Prompt și deschideți-l. Aici vom rula comenzile.

Odată ce programul este deschis, vom crea un mediu virtual pentru instalarea noastră PyTorch folosind comanda.

conda create -n pytorch python=3.7

După crearea mediului virtual, îl putem activa rulând următoarea comandă:

conda activate pytorch

Odată ce mediul virtual este activ, putem trece la instalarea PyTorch. Începem prin a accesa pagina de instalare a site-ului web PyTorch. După care, putem derula în jos până la secțiunea din pagină în care se află acest asistent de instalare:

Aici, selectăm informațiile despre sistem, iar expertul ne va da o comandă pentru a instala PyTorch. Voi selecta versiunea stabilă pentru Windows, gestionată de Conda, utilizată prin limbajul de programare Python și rulează pe un procesor. Ca rezultat, masa mea va arăta așa.

Apoi, copiați comanda, lipiți-o în promptul Anaconda și apăsați ENTER.

După finalizarea instalării, putem verifica dacă a avut succes deschizând shell-ul interactiv Python și încercând să importăm PyTorch.

Deci, în interiorul promptului Anaconda, începeți o sesiune interactivă Python.

python

După începerea sesiunii, importați PyTorch folosind următoarea linie de cod:

import torch

Dacă această acțiune se finalizează fără erori, atunci instalarea a avut succes.

Cuvinte finale

În acest ghid, am instalat PyTorch atât pe Windows, cât și pe Linux, folosind conda. Este posibil să îl instalați prin PIP ca un pachet PIP normal. In ambele cazuri am optat pentru instalarea CPU. Cu toate acestea, puteți utiliza în continuare CUDA, care este un set de instrumente de sistem dezvoltat de Nvidia care accelerează antrenamentul prin paralelizarea operațiunilor pe GPU-uri.

  6 cele mai bune jocuri din magazinul Windows