Установка PyTorch на Linux Mint может быть выполнена с помощью pip. В этом примере я покажу, как установить PyTorch с помощью pip.

Шаг 1: Установка Python и pip

Убедитесь, что у вас установлен Python и pip. Вы можете проверить это, выполнив следующие команды в терминале:

python3 --version
pip3 --version

Если Python или pip не установлены, вы можете установить их с помощью следующей команды:

sudo apt update
sudo apt install python3 python3-pip

Практически к каждому материалу есть видеоролик на  YouTube  и  RUTUBE каналах, теорию можно прочитать в статье, а как работает устройство в железе, посмотреть в видеоролике! 

Шаг 2: Установка PyTorch
Обновите систему и установите необходимые зависимости:

sudo apt update
sudo apt upgrade
sudo apt install python3-pip python3-dev

Установите виртуальное окружение (рекомендуется):

sudo apt update
sudo apt install python3-pip python3-venv

Создайте виртуальное окружение:

python3 -m venv pytorch_env

Активируйте виртуальное окружение:

source pytorch_env/bin/activate

После активации виртуального окружения ваш терминал должен показать имя окружения в начале строки.
Установите PyTorch:

Перейдите на официальный сайт PyTorch — https://pytorch.org/get-started/locally/ — там можно выбрать нужную конфигурацию (версию PyTorch, CUDA или CPU).

Например, для установки последней версии PyTorch с поддержкой CPU выполните:

pip install torch torchvision torchaudio

Если у вас есть видеокарта NVIDIA и вы хотите использовать CUDA (например, CUDA 11.8), команда будет примерно такой:

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

Проверьте установку:

Запустите Python и попробуйте импортировать  PyTorch: 

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # True, если CUDA доступна

Понравился проект? Не забудь поделиться им с друзьями в соц. сетях.

А также подписаться на наш канал на YouTube и RUTUBE !

• СПАСИБО ЗА ПРОСМОТР.
• С уважением.
• Электроника и Робототехника!