Î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.
Cuprins
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.
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.
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.