Ga naar inhoud

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.

  1. Installeer Python:

    sudo apt install python3.10
    

    In Ubuntu 22.04 is deze versie standaard geïnstalleerd, dus we raden niet aan een nieuwere versie te installeren.

  2. Maak een virtuele omgeving voor Python:

    python3 -m venv venv
    
  3. Activeer de virtuele omgeving:

    source venv/bin/activate
    

    Na succesvolle activatie bevat de prompt de naam van de virtuele omgeving tussen haakjes:

    (venv) user@49069:~$
    

    Opmerking

    U kunt zoveel virtuele omgevingen maken als u wilt en verschillende bibliotheken installeren (inclusief gelijktijdig, maar soms kan dit conflicten veroorzaken).

  4. Installeer PyTorch-bibliotheken in de virtuele omgeving:

    pip3 install torch torchvision torchaudio
    
  5. Verifieer de PyTorch-installatie:

    Voer hiervoor een Python-consolecommando python uit en voer vervolgens het volgende programma uit:

    import torch
    x = torch.rand(5, 3)
    print(x)
    

    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]])
    
  6. Verifieer of PyTorch-bibliotheken CUDA gebruiken:

    Voer hiervoor het volgende programma uit in een Python-console:

    import torch
    torch.cuda.is_available()
    

    Als PyTorch in GPU-modus kan werken, is de uitvoer:

     >>> import torch
     >>> torch.cuda.is_available()
    True
    

Om de virtuele omgeving te verlaten, gebruikt u het commando:

deactivate
question_mark
Is there anything I can help you with?
question_mark
AI Assistant ×