12.12.2024

Jak zainstalować PyTorch na AlmaLinux

Instalacja PyTorch na AlmaLinux jest prostym procesem, choć wymaga pewnej znajomości wiersza poleceń Linuksa i zarządzania pakietami Python. W tym przewodniku przejdziemy przez kroki, aby uruchomić PyTorch na komputerze z systemem AlmaLinux. Pod koniec tego samouczka PyTorch będzie zainstalowany i gotowy do użycia w projektach uczenia maszynowego lub głębokiego uczenia.

Wymagania wstępne

Przed rozpoczęciem instalacji należy upewnić się, że

Na komputerze zainstalowany jest system AlmaLinux.
Masz uprawnienia sudo.
Zainstalowany jest Python 3.8 lub nowszy.
zainstalowany jest pip do zarządzania pakietami Pythona.

Jeśli nie masz zainstalowanego Pythona lub pip, możesz je zainstalować za pomocą poniższych poleceń:

sudo dnf install python3
sudo dnf install python3-pip

Teraz przejdźmy do instalacji PyTorch.

Krok 1: Aktualizacja pakietów systemowych

Po pierwsze, upewnij się, że pakiety systemowe są aktualne, aby uniknąć problemów z kompatybilnością podczas instalacji.

sudo dnf update -y

To polecenie zaktualizuje wszystkie zainstalowane pakiety do ich najnowszych wersji.

Krok 2: Zainstaluj narzędzia programistyczne Python

Potrzebujesz narzędzi programistycznych Python i virtualenv, aby stworzyć izolowane środowisko Python dla PyTorch.

sudo dnf groupinstall “Narzędzia deweloperskie”
sudo dnf install python3-devel
sudo pip3 install virtualenv

Krok 3: Utwórz środowisko wirtualne (opcjonalnie)

Utworzenie wirtualnego środowiska jest opcjonalne, ale zalecane. Pomaga ono utrzymać porządek w projektach Pythona i uniknąć konfliktów pakietów.

mkdir pytorch_env
cd pytorch_env
python3 -m venv venv
źródło venv/bin/activate

Teraz terminal powinien pokazywać, że pracujesz w środowisku wirtualnym ((venv) powinno pojawić się przed wierszem zachęty).

Krok 4: Instalacja PyTorch przy użyciu pip

Aby zainstalować PyTorch, możesz użyć oficjalnego polecenia instalacji PyTorch. Odwiedź stronę PyTorch i wybierz odpowiednie opcje (takie jak kompilacja PyTorch, system operacyjny, menedżer pakietów i wersja CUDA). Tutaj założymy, że chcesz zainstalować najnowszą stabilną wersję bez GPU (wersja tylko na CPU):

pip install torch torchvision torchaudio

Jeśli masz procesor graficzny kompatybilny z CUDA i chcesz go wykorzystać w PyTorch, musisz zainstalować wersję odpowiadającą wersji CUDA. Na przykład, jeśli masz zainstalowaną wersję CUDA 11.7:

pip install torch torchvision torchaudio –index-url https://download.pytorch.org/whl/cu117

Upewnij się, że zainstalowałeś odpowiednią wersję CUDA w swoim systemie AlmaLinux przed użyciem tego polecenia.

Krok 5: Weryfikacja instalacji

Po zakończeniu instalacji możesz sprawdzić, czy PyTorch jest poprawnie zainstalowany, uruchamiając Pythona i uruchamiając prosty skrypt.

python

Następnie w interaktywnej powłoce Pythona wpisz:

import torch
print(torch.__version__)
print(torch.cuda.is_available())

Wynik powinien wyświetlać zainstalowaną wersję PyTorch. Jeśli zainstalowałeś wersję kompatybilną z CUDA i masz GPU obsługujący CUDA, torch.cuda.is_available() powinien zwrócić True.

Krok 6: Dezaktywacja środowiska wirtualnego (jeśli jest używane)

Po zweryfikowaniu instalacji można dezaktywować środowisko wirtualne:

deactivate

Aby ponownie aktywować środowisko w przyszłości, przejdź do katalogu projektu i użyj:

source venv/bin/activate

Wnioski

PyTorch został pomyślnie zainstalowany na AlmaLinux! Teraz możesz zacząć budować i uruchamiać swoje modele głębokiego uczenia za pomocą PyTorch. Postępując zgodnie z tym przewodnikiem, dowiedziałeś się również, jak utworzyć wirtualne środowisko Python, co jest przydatną umiejętnością do zarządzania projektami Python.

Miłego kodowania z PyTorch! Jeśli napotkasz jakiekolwiek problemy, zapoznaj się z dokumentacją PyTorch, aby uzyskać bardziej szczegółowe instrukcje i wskazówki dotyczące rozwiązywania problemów.