I denne vejledning vil jeg gennemgå installationen af PyTorch på Windows og Linux ved at bruge Anaconda til at styre installationen af pakkerne.
Installationen vil blive udført ved hjælp af kommandolinjeværktøjet conda, der er indbygget i Anaconda.
PyTorch er et maskinlæringsbibliotek skrevet i Python og er baseret på Torch-rammen. Det er udviklet af Facebook og kan sammenlignes med Tensorflow af Google. Det er nyttigt inden for computersyn og naturlig sprogbehandling og er blevet brugt af virksomheder som Tesla til at udvikle autopilotsoftware.
PyTorch er gratis og open source, licenseret under den modificerede BSD og er under Linux Foundation.
Indholdsfortegnelse
Forudsætninger
For at følge denne vejledning skal du have Anaconda installeret på den maskine, du arbejder på.
Hvis du ikke allerede har det installeret, vil denne guide til, hvordan du installerer Anaconda, guide dig gennem hele processen. Efter at have fulgt denne vejledning, kan du fortsætte med at installere PyTorch.
Installation af PyTorch på Linux
Som en god praksis skal du begynde med at opdatere softwarepakker i din Linux-distribution. I mit tilfælde bruger jeg Ubuntu og er egnet til at administrere mine pakker, så jeg vil bruge følgende kommando til at opdatere:
$ sudo apt update && apt upgrade
Når du er færdig med at opdatere pakkerne, skal du gå til den officielle installationsside for PyTorch-webstedet. Rul ned på siden, indtil du finder installationsguiden, der ser ud som nedenstående:
Ved hjælp af denne guide vil du være i stand til at klikke på de forskellige muligheder for at give dine systemoplysninger og præferencer, og til gengæld får du en kommando, du kan bruge i din terminal til at installere PyTorch.
Efter at have givet mine systemoplysninger, ser det sådan ud:
Jeg valgte den stabile Linux-version, og jeg vil bruge Conda til at administrere mine pakker. Jeg valgte også at bruge PyTorch med Python frem for C++/Java. Og jeg vil køre min PyTorch på en CPU i modsætning til en GPU.
Nederst i tabellen er den kommando, jeg kan bruge til at installere PyTorch, men før jeg kører denne kommando, vil jeg gerne oprette et virtuelt Anaconda-miljø kaldet pytorch.
Virtuelle miljøer giver dig mulighed for at oprette projekter og holde deres afhængigheder isoleret fra andre projekters afhængigheder og dermed forhindre afhængighedskonflikter. En af fordelene ved Anaconda er, at det hjælper dig med nemt at skabe og administrere virtuelle miljøer.
For at skabe et virtuelt miljø, hvor Python-versionen er 3.7, vil jeg indtaste følgende kommando:
conda create -n pytorch python=3.7
Når miljøet er oprettet, aktiverer jeg det ved hjælp af følgende kommando:
conda activate pytorch
Når miljøet er aktivt, vil jeg køre kommandoen genereret tidligere på PyTorch-webstedet for at installere PyTorch.
conda install pytorch torchvision torchaudio cpuonly -c pytorch
Følg vejledningen for at installere PyTorch. Når det er gjort, genstarter jeg terminalsessionen for at træde i kraft.
For nu at bekræfte, at PyTorch var installeret korrekt, vil vi prøve at importere det i den interaktive Python-skal. Sørg for, at du er i det virtuelle Pytorch-miljø ved hjælp af kommandoen:
conda activate pytorch
Når du er i det virtuelle Pytorch-miljø, skal du åbne den interaktive python-skal ved at skrive kommandoen:
python
Når shell-sessionen starter, skal du skrive følgende kodelinje og trykke på ENTER
import torch
Hvis Python kører uden fejl, var installationen vellykket. Men hvis du får fejlen Modul ikke fundet, betyder det, at noget gik galt under installationen. Du kan prøve at geninstallere det igen.
Installation af PyTorch på Windows
For at begynde skal du på din Windows-maskine søge efter Anaconda Prompt-programmet og åbne det. Det er her, vi skal køre kommandoerne.
Når programmet er åbent, skal vi oprette et virtuelt miljø til vores PyTorch-installation ved hjælp af kommandoen.
conda create -n pytorch python=3.7
Efter at have oprettet det virtuelle miljø, kan vi aktivere det ved at køre følgende kommando:
conda activate pytorch
Når det virtuelle miljø er aktivt, kan vi fortsætte med at installere PyTorch. Vi begynder med at gå til PyTorch-webstedets installationsside. Hvorefter vi kan rulle ned til den del af siden, hvor denne installationsguide er placeret:
Her vælger vi vores systemoplysninger, og guiden vil give os en kommando til at installere PyTorch. Jeg vil vælge den stabile udgivelse til Windows, administreret af Conda, brugt gennem programmeringssproget python og kører på en CPU. Som et resultat kommer mit bord til at se sådan ud.
Kopier derefter kommandoen, indsæt den i Anaconda-prompten og tryk på ENTER.
Når installationen er fuldført, kan vi bekræfte, om den var vellykket, ved at åbne den interaktive Python-skal og prøve at importere PyTorch.
Så inde i Anaconda-prompten, start en interaktiv Python-session.
python
Når sessionen starter, importerer du PyTorch ved hjælp af følgende kodelinje:
import torch
Hvis denne handling fuldføres uden fejl, var installationen vellykket.
Afsluttende ord
I denne vejledning installerede vi PyTorch på både Windows og Linux ved hjælp af conda. Det er muligt at installere det via PIP som en normal PIP-pakke. I begge tilfælde valgte jeg CPU-installationen. Du kan dog stadig bruge CUDA, som er et systemværktøjssæt udviklet af Nvidia, der fremskynder træningen ved at parallelisere operationer på tværs af GPU’er.