12.12.2024

Як встановити PyTorch на AlmaLinux

Встановлення PyTorch на AlmaLinux є простим процесом, хоча і вимагає певних знань командного рядка Linux та керування пакунками Python. У цьому посібнику ми розглянемо кроки для запуску PyTorch на машині з AlmaLinux. Наприкінці цього підручника ви встановите PyTorch і будете готові використовувати його у своїх проектах машинного навчання або глибокого навчання.

Необхідні умови

Перед початком встановлення переконайтеся у наступному:

На вашому комп’ютері встановлено AlmaLinux.
Ви маєте привілеї sudo.
Встановлено Python 3.8 або новішої версії.
встановлено pip для керування пакунками Python.

Якщо у вас не встановлено Python або pip, ви можете встановити їх за допомогою наступних команд:

sudo dnf install python3
sudo dnf install python3-pip

Тепер перейдемо до встановлення PyTorch.

Крок 1: Оновлення системних пакетів

Перш за все, переконайтеся, що ваші системні пакунки оновлені, щоб уникнути проблем сумісності під час встановлення.

sudo dnf update -y

Ця команда оновить усі встановлені пакунки до найновіших версій.

Крок 2: Встановлення інструментів розробки Python

Для створення ізольованого середовища Python для PyTorch вам знадобляться інструменти розробки Python та virtualenv.

sudo dnf groupinstall “Development Tools”
sudo dnf install python3-devel
sudo pip3 install virtualenv

Крок 3: Створення віртуального середовища (необов’язково)

Створення віртуального середовища не є обов’язковим, але рекомендується. Воно допомагає впорядкувати ваші Python-проекти та уникнути конфліктів між пакунками.

mkdir pytorch_env
cd pytorch_env
python3 -m venv venv
джерело venv/bin/activate

Тепер ваш термінал повинен показати, що ви працюєте у віртуальному середовищі ((venv) має з’явитися перед запрошенням).

Крок 4: Встановлення PyTorch за допомогою pip

Щоб встановити PyTorch, ви можете скористатися офіційною командою встановлення PyTorch. Відвідайте веб-сайт PyTorch і виберіть відповідні параметри (наприклад, збірку PyTorch, ОС, менеджер пакунків і версію CUDA). Тут ми будемо вважати, що ви хочете встановити останню стабільну версію без GPU (версія тільки для CPU):

pip install torch torchvision torchaudio

Якщо ви маєте CUDA-сумісний графічний процесор і хочете використовувати його для PyTorch, вам потрібно встановити версію, яка відповідає вашій версії CUDA. Наприклад, якщо у вас встановлено CUDA 11.7:

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

Переконайтеся, що у вашій системі AlmaLinux встановлено відповідну версію CUDA, перш ніж використовувати цю команду.

Крок 5: Перевірка встановлення

Після завершення встановлення ви можете перевірити правильність встановлення PyTorch, запустивши Python і виконавши простий скрипт.

python

Потім в інтерактивній оболонці Python введіть

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

На виході має бути виведено встановлену версію PyTorch. Якщо ви встановили CUDA-сумісну версію і маєте графічний процесор з підтримкою CUDA, torch.cuda.is_available() має повернути True.

Крок 6: Вимкнення віртуального середовища (якщо використовується)

Після перевірки встановлення ви можете вимкнути віртуальне середовище:

деактивувати

Щоб активувати середовище у майбутньому, перейдіть до каталогу проекту і скористайтеся ним:

source venv/bin/activate

Висновок

Ви успішно встановили PyTorch на AlmaLinux! Тепер ви можете почати створювати та запускати свої моделі глибокого навчання за допомогою PyTorch. Дотримуючись цього посібника, ви також навчилися створювати віртуальне середовище Python, що є корисною навичкою для керування проектами Python.

Вдалого кодування з PyTorch! Якщо у вас виникнуть проблеми, обов’язково зверніться до документації PyTorch для отримання більш детальних інструкцій та порад щодо усунення несправностей.