PyTorch – один з найпопулярніших фреймворків машинного навчання з відкритим вихідним кодом, розроблений лабораторією AI Research Facebook. Він широко використовується для додатків глибокого навчання, включаючи комп’ютерний зір і обробку природної мови. Якщо ви використовуєте AlmaLinux, стабільний і надійний дистрибутив Linux на основі RHEL, цей посібник покаже вам, як швидко та ефективно встановити PyTorch.

Передумови

Перш ніж почати, переконайтеся, що

  • Ви маєте працюючий екземпляр AlmaLinux 8 або 9

  • Ви маєте привілеї root або sudo

  • Встановлено Python 3.7 або новішої версії

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

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

sudo dnf update -y

За бажанням, встановіть EPEL для доступу до додаткових пакунків:

sudo dnf install epel-release -y

Крок 2: Встановіть Python та pip

Перевірте, чи Python вже встановлено:

python3 --version

Якщо його не встановлено, ви можете встановити його за допомогою:

sudo dnf install python3 python3-pip -y

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

Щоб тримати ваше середовище Python чистим та ізольованим, варто створити віртуальне середовище:

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

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

Щоб встановити PyTorch з підтримкою лише процесора:

pip install torch torchvision torchaudio torchvision torchaudio

Якщо вам потрібне прискорення графічного процесора і у вас є підтримуваний графічний процесор NVIDIA, а також встановлені відповідні драйвери і CUDA, скористайтеся відповідною командою. Наприклад, для встановлення PyTorch з підтримкою CUDA 11.8:

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

Найсвіжіші варіанти встановлення можна знайти на офіційному сайті PyTorch:
https://pytorch.org/get-started/locally

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

Щоб перевірити правильність встановлення PyTorch, запустіть:

python

Потім введіть наступне в оболонці Python:

import torch
print(torch.__version__)
print("CUDA available:", torch.cuda.is_available())

Якщо все працює, ви побачите встановлену версію PyTorch і чи доступна CUDA.

Поради з усунення несправностей

  • Якщо у вас виникли проблеми з pip або SSL-сертифікатами, оновіть pip:

pip install --upgrade pip

  • Якщо ви використовуєте CUDA, переконайтеся, що драйвери NVIDIA та інструментарій CUDA встановлено правильно і вони сумісні з версією PyTorch, яку ви встановлюєте.

Висновок

Ви успішно встановили PyTorch на AlmaLinux. Незалежно від того, чи ви розробляєте моделі глибокого навчання, чи проводите експерименти, PyTorch надає гнучкий та потужний фреймворк для машинного навчання на Linux-системах. Для більш просунутих налаштувань розгляньте можливість інтеграції з Jupyter Notebook або використання контейнерів Docker.