Delen via

PyTorch installeren en configureren

In de vorige fase van deze zelfstudie hebben we de basisbeginselen van PyTorch en de vereisten besproken voor het maken van een machine learning-model. Hier installeren we deze op uw computer.

PyTorch ophalen

Eerst moet u een Python-omgeving instellen.

Het is raadzaam om een virtuele Python-omgeving in Windows in te stellen, waarbij Anaconda wordt gebruikt als pakketbeheerder. In de rest van deze installatie wordt ervan uitgegaan dat u een Anaconda-omgeving gebruikt.

  1. Download en installeer Anaconda hier. Selecteer Anaconda 64-bit installer for Windows Python 3.8.

Belangrijk

Houd er rekening mee dat u Python 3.x installeert. PyTorch in Windows ondersteunt momenteel alleen Python 3.x; Python 2.x wordt niet ondersteund.

Anaconda-installatieprogramma's

Nadat de installatie is voltooid, controleert u uw Anaconda- en Python-versies.

  1. Open Anaconda Manager via Start - Anaconda3 - Anaconda PowerShell Prompt en test uw versies:

U kunt uw Python-versie controleren door de volgende opdracht uit te voeren: python –-version

U kunt uw Anaconda-versie controleren door de volgende opdracht uit te voeren: conda –-version

Opdrachtregelprompt anaconda

U kunt nu het PyTorch-pakket installeren vanuit binaire bestanden via Conda.

  1. Navigeer naar https://pytorch.org/.

Selecteer de relevante PyTorch-installatiedetails:

  • PyTorch build – stabiel.
  • Uw besturingssysteem - Windows
  • Pakket – Conda
  • Taal - Python
  • Compute Platform : CPU of kies uw versie van Cuda. In deze zelfstudie traint u een model en voert u inferentie uit op een CPU, maar u kunt ook een Nvidia GPU gebruiken.
  1. Open Anaconda-beheer en voer de opdracht uit zoals deze is opgegeven in de installatie-instructies.
conda install pytorch torchvision torchaudio cpuonly -c pytorch

Anaconda-installatie

  1. Bevestig en voltooi de extractie van de vereiste pakketten.

Anaconda-installatie bevestigen

We gaan de installatie van PyTorch controleren door voorbeeldcode van PyTorch uit te voeren om een willekeurig geïnitialiseerde tensor te maken.

  1. Open de Anaconda PowerShell-prompt en voer de volgende opdracht uit.
python

Voer vervolgens de volgende code in:

import torch 

x = torch.rand(2, 3) 

print(x)

De uitvoer moet een willekeurige tensor van 5x3 zijn. De getallen zijn anders, maar deze moeten er ongeveer als volgt uitzien.

Pytorch-functionaliteit controleren

Opmerking

Bent u geïnteresseerd in meer informatie? Ga naar de officiële website van PyTorch

Volgende stappen

Nu we PyTorch hebben geïnstalleerd, zijn we klaar om de gegevens voor ons model in te stellen.

  • Last updated on