PyTorch-installatie¶
PyTorch is een framework voor de programmeertaal Python, ontworpen voor machine learning. Het bevat een set tools voor het werken met modellen, die veel worden gebruikt in natuurlijke taalverwerking, computer vision en andere vergelijkbare domeinen. U kunt het handmatig op uw server installeren met behulp van deze handleiding.
PyTorch-installatie op Linux¶
Deze instructie is geschikt voor de volgende besturingssystemen: Ubuntu 22.04, en geverifieerd voor Python-versies: Python 3.10.
Opmerking
Als u van plan bent GPU-versnelling te gebruiken, installeer dan NVIDIA-stuurprogramma's en CUDA volgens deze instructie.
-
Installeer Python:
In Ubuntu 22.04 is deze versie standaard geïnstalleerd, dus we raden niet aan een nieuwere versie te installeren.
-
Maak een virtuele omgeving voor Python:
-
Activeer de virtuele omgeving:
Na succesvolle activatie bevat de prompt de naam van de virtuele omgeving tussen haakjes:
Opmerking
U kunt zoveel virtuele omgevingen maken als u wilt en verschillende bibliotheken installeren (inclusief gelijktijdig, maar soms kan dit conflicten veroorzaken).
-
Installeer PyTorch-bibliotheken in de virtuele omgeving:
-
Verifieer de PyTorch-installatie:
Voer hiervoor een Python-consolecommando
pythonuit en voer vervolgens het volgende programma uit:Als de installatie succesvol is, ontvangt u de volgende uitvoer:
(venv) user@49069:~$ python Python 3.10.12 (main, Nov 20 2023, 15:14:05) [GCC 11.4.0] on linux Type "help", "copyright", "credits" or "license" for more information. >>> import torch >>> x = torch.rand(5, 3) >>> print(x) tensor([[0.80, 0.04, 0.6], [0.32, 0.59, 0.7], [0.8, 0.70, 0.25], [0.40, 0.9, 0.9], [0.8, 0.15, 0.5]]) -
Verifieer of PyTorch-bibliotheken CUDA gebruiken:
Voer hiervoor het volgende programma uit in een Python-console:
Als PyTorch in GPU-modus kan werken, is de uitvoer:
Om de virtuele omgeving te verlaten, gebruikt u het commando: