Как установить Pytorch на Ubuntu 22.04? (Пошаговое руководство по установке Pytorch в Ubuntu)

PyTorch - это широко используемый фреймворк машинного обучения с открытым исходным кодом, который отличается простотой использования, динамическим графиком вычислений и эффективностью, что делает его идеальным для задач глубокого обучения.

Разработанный компанией Meta AI, которая теперь является частью Linux Foundation, PyTorch используется для распознавания изображений и речи, обработки естественного языка и обучения с подкреплением. его мощная архитектура, оптимизированная для графических процессоров, позволяет аналитикам данных легко создавать и развертывать нейронные сети, что делает его важным инструментом для исследований и разработок в области ИИ.

Это руководство поможет вам установить PyTorch на Ubuntu с помощью Pip или Anaconda для начала работы.

Необходимые условия

Убедитесь, что ваша система соответствует этим требованиям:

Python 3.6 или более поздней версии
Для поддержки графического процессора убедитесь, что установлены драйверы CUDA и инструментарий.

Установка PyTorch на Ubuntu (Установка PyTorch Ubuntu)

Для установки PyTorch в систему Ubuntu с поддержкой CPU или GPU вы можете использовать Pip или Anaconda. Мы сосредоточимся на том, как установить Pytorch с помощью Pip и настроить виртуальную среду для упорядочения ваших пакетов Python.

Способ 01: установка PyTorch на Ubuntu с помощью Pip3 (установка Pytorch Ubuntu)

Вы можете легко установить PyTorch на Ubuntu с помощью Pip (менеджер пакетов Python), выполнив следующие шаги:

Шаг 1: Обновление системных пакетов

Во-первых, убедитесь, что ваши системные пакеты обновлены. Откройте терминал и выполните следующую команду:

$ sudo apt update

Это обновит ваши списки пакетов, чтобы убедиться, что у вас есть последние версии.

Шаг 2: Установка Python3-venv

Затем вам нужно установить python3-venv, который позволяет создавать изолированные среды Python. Это важно для предотвращения конфликтов между проектами, которым требуются разные версии пакетов.

$ sudo apt install python3-venv -y

Шаг 3: Настройка виртуальной среды Python

Чтобы настроить виртуальную среду Python, создайте каталог для вашего проекта Python и перейдите в него с помощью команды cd. Эта директория будет содержать виртуальную среду и связанные с ней файлы. Например, мы создали каталог с названием « pytorch_env».

$ mkdir pytorch_env
$ cd pytorch_env

Теперь создадим и активируем виртуальную среду:

python3 -m venv pytorch_env
source pytorch_env/bin/activate

После активации виртуальной среды в вашем терминале изменится надпись, которая покажет, что вы работаете в среде pytorch_env. Теперь вы можете устанавливать пакеты Python или запускать скрипты именно в этой среде. Это помогает избежать конфликтов с другими проектами, поскольку зависимости не зависят от общесистемной установки Python.

Шаг 4: Установка PyTorch с помощью Pip (установка Pytorch Ubuntu с помощью Pip)

Когда виртуальная среда активирована, вы можете установить PyTorch. Если у вас нет графического процессора или вам не нужно ускорение графического процессора. Вы можете установить библиотеку Pytorch только для поддержки CPU. Чтобы установить Pytorch для работы на CPU, используйте эту команду:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

Если у вас есть совместимый графический процессор и вы хотите использовать CUDA для ускорения, установите PyTorch с поддержкой GPU:

pip3 install torch torchvision torchaudio

Шаг 5: Проверка установки

Чтобы убедиться, что PyTorch был успешно установлен, откройте интерпретатор Python и проверьте версию PyTorch:

python
import torch

value = torch.randn(1, 6)

print(torch.__version__)

Это выведет версию PyTorch, которую вы установили, подтверждая, что установка прошла успешно.

Способ 02: Как установить PyTorch в Ubuntu с помощью Anaconda

Вы можете установить PyTorch на Ubuntu с помощью Anaconda, популярной платформы с открытым исходным кодом. Этот метод позволяет легко управлять пакетами и окружением Python. Чтобы установить Pytorch с помощью Conda, выполните следующие шаги:

Шаг 1: Обновление системных пакетов

Начните с обновления системных пакетов, чтобы убедиться, что они актуальны. Откройте терминал и выполните следующие действия

$ sudo apt update

Шаг 2: Установка Anaconda или Conda

Затем вам нужно установить Anaconda. Пакеты Conda можно установить с помощью команды Curl. Если curl еще не установлен в вашей системе Ubuntu, установите его с помощью следующей команды:

$ sudo apt install curl -y

Перейдите во временный каталог: Рекомендуется выполнять установку в изолированном каталоге. Вы можете использовать каталог /tmp:

cd /tmp

Загрузите скрипт установщика Anaconda с помощью Curl для загрузки установщика:

curl --output anaconda.sh

https://repo.anaconda.com/archive/Anaconda3-2023.07-2-Linux-x86_64.sh

Эта команда сохранит установщик Anaconda как «anaconda.sh». При необходимости вы можете проверить последнюю версию на официальном сайте Anaconda.

Проверьте загруженный файл. Важно проверить целостность загруженного файла, чтобы убедиться, что он не был поврежден. Проверьте его с помощью

sha256sum anaconda.sh

Убедитесь, что контрольная сумма совпадает со значением, указанным на официальном сайте Anaconda.

После проверки продолжайте установку, запустив установочный скрипт:

bash anaconda.sh

После установки обновите сеанс командной оболочки, чтобы сделать команды Anaconda доступными в вашем терминале:

source ~/.bashrc

Вы должны увидеть, что строка терминала изменилась на «base», что указывает на то, что среда Anaconda активирована по умолчанию. Позже вы можете активировать Anaconda с помощью команды conda activate.

Убедитесь, что Anaconda установлена, запустив ее:

conda --version

Шаг 3: Установка PyTorch с помощью Anaconda

После установки и активации Anaconda вы можете установить PyTorch.

Если в вашей системе нет графического процессора или вам не нужно ускорение, установите PyTorch с поддержкой CPU:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

Если у вас есть графический процессор и вы хотите использовать CUDA для ускорения, установите PyTorch с поддержкой GPU:

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

Шаг 4: проверка установки

Наконец, чтобы убедиться, что PyTorch был установлен правильно, запустите сеанс Python и проверьте версию PyTorch:

python
import torch
print(torch.__version__)

Это выведет версию PyTorch, которая установлена на данный момент, подтверждая, что все настроено правильно.

Как удалить PyTorch в Ubuntu?

Чтобы удалить PyTorch из вашей системы Ubuntu, выполните эти шаги в зависимости от того, как вы его установили.

Если вы установили PyTorch с помощью Pip, вы можете удалить его, выполнив следующую команду:

$ pip3 uninstall torch -y

Эта команда удалит PyTorch из вашей среды Python.

Если вы использовали Anaconda для установки PyTorch, вы можете удалить его с помощью этой команды:

$ conda remove pytorch -y

Это удалит PyTorch из вашей среды Anaconda.

Заключение

В этом руководстве было продемонстрировано, как установить PyTorch в Ubuntu с помощью Pip и Anaconda. Он также рассмотрел различные варианты установки, доступные в зависимости от вычислительных ресурсов вашей системы. Теперь, когда PyTorch настроен, вы можете погрузиться в его функции и воспользоваться его мощными возможностями.

Веб-хостинг VPS - это мощный и гибкий вариант, который может существенно повлиять на успех вашего сайта. С blueVPS вы можете арендовать серверы по всему миру, включая cPanel для удобного управления. Готовы повысить производительность своего сайта? Подумайте о VPS-хостинге уже сегодня.

Blog