Может ли мой ПК поддерживать и как установить ROCm 6.1?

Содержание

Вопрос или проблема
Ответ или решение
Как установить ROCm 6.1 на ваш компьютер с Ubuntu
Шаг 1: Проверка совместимости
Шаг 2: Удаление предыдущих версий ROCm
Шаг 3: Добавление репозитория ROCm
Шаг 4: Обновление системы и установка ROCm 6.1
Шаг 5: Проверка установки ROCm
Шаг 6: Установка PyTorch 2.4 с ROCm 6.1
Заключение

Вопрос или проблема

Я пытаюсь изучить и запустить код машинного обучения на своем компьютере.

Я хотел бы установить PyTorch 2.4, и у него есть возможность установить его с Compute Platform ROCm 6.1. Я полагаю, у меня есть ROCm, близкий к 6.1, но не совсем. Могу ли я и как я могу обновить до ROCm 6.1 на своем Ubuntu?

Для справки, у моего устройства следующие характеристики:


    Модель оборудования: Lenovo ThinkPad X13 Gen 4
    …
    Процессор: AMD Ryzen™ 7 PRO 7840U с графикой Radeon™ 780M × 16
    Графика: AMD Radeon™ Graphics
    Версия прошивки: R29ET56W (1.30)
    …
    Название ОС: Ubuntu 24.04.1 LTS
    Сборка ОС: (null)
    Тип ОС: 64-бит
    Версия GNOME: 46
    Система окон: Wayland
    Версия ядра: Linux 6.8.0-47-generic

Похоже, у меня есть более старая версия(и) ROCm. Когда я использую “apt show rocm…” в командной строке, я получаю:

1.)
Пакет: rocminfo
Версия: 5.7.1-3build1
Приоритет: необязательный
Раздел: universe/devel
Происхождение: Ubuntu
Установленный размер: 95.2 кБ
Зависимости: libc6 (>= 2.34), libgcc-s1 (>= 3.3.1), libhsa-runtime64-1 (>= 5.7.1~), libstdc++6 (>= 11), python3, pciutils, kmod
Размер загрузки: 25.6 кБ

2.)
Пакет: rocm-cmake
Версия: 6.0.0-1
Приоритет: необязательный
Раздел: universe/devel
Происхождение: Ubuntu
Установленный размер: 138 кБ
Размер загрузки: 25.6 кБ

3.)
Пакет: rocm-device-libs-17
Версия: 6.0+git20231212.5a852ed-2
Приоритет: необязательный
Раздел: universe/libs
Источник: rocm-device-libs
Происхождение: Ubuntu
Установленный размер: 3,324 кБ
Размер загрузки: 549 кБ

4.)
Пакет: rocm-smi
Версия: 5.7.0-1
Приоритет: необязательный
Раздел: universe/utils
Источник: rocm-smi-lib
Происхождение: Ubuntu
Установленный размер: 227 кБ
Зависимости: python3:any, librocm-smi64-1 (= 5.7.0-1)
Размер загрузки: 52.9 кБ

Ни одна из версий не соответствует 6.1 или выше, так что я думаю, что мне стоит обновить перед установкой, да? Я видел некоторую информацию здесь, но не понял ее достаточно, чтобы успешно установить. Есть ли кто-то, кто знаком с обновлением ROCm и может помочь мне здесь? 🙂

Ответ или решение

Как установить ROCm 6.1 на ваш компьютер с Ubuntu

Ваше оборудование, Lenovo ThinkPad X13 Gen 4 с процессором AMD Ryzen™ 7 PRO 7840U и графикой Radeon™, подходит для установки ROCm. На основе предоставленной вами информации, вы действительно используете более старую версию ROCm и должны обновить её до версии 6.1, чтобы иметь возможность работать с PyTorch 2.4 и использовать поддержку ROCm.

Шаг 1: Проверка совместимости

Перед началом убедитесь, что ваше оборудование соответствует минимальным требованиям для ROCm 6.1. Система должна использовать:

Операционная система: Ubuntu 20.04 или более поздняя версия (вы используете Ubuntu 24.04.1 LTS, что является отличным выбором).
Процессор: Подходит, так как у вас AMD Ryzen 7.
Графическая карта: Убедитесь, что графический процессор соответствует требованиям ROCm 6.1.

Шаг 2: Удаление предыдущих версий ROCm

Перед установкой ROCm 6.1 рекомендуется удалить устаревшие версии ROCm. Используйте следующие команды в терминале:

sudo apt-get remove --purge rocm*

Это позволит удалить все установленные ранее пакеты ROCm.

Шаг 3: Добавление репозитория ROCm

Следующим шагом будет добавление официального репозитория ROCm для Ubuntu. Выполните следующие команды:

echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/debian/ $(lsb_release -sc) main' | sudo tee /etc/apt/sources.list.d/rocm.list

Импорт ключа репозитория:

wget -qO - https://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | sudo apt-key add -

Шаг 4: Обновление системы и установка ROCm 6.1

Теперь обновите список пакетов и установите ROCm 6.1:

sudo apt update
sudo apt install rocm-dkms

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

После установки ROCm проверьте, что он установлен корректно. Выполните в терминале следующую команду:

/opt/rocm/bin/rocminfo

Это должно вывести информацию о вашей графической карте и установленной версии ROCm.

Шаг 6: Установка PyTorch 2.4 с ROCm 6.1

Теперь, когда ROCm 6.1 установлен, вы можете установить PyTorch с поддержкой ROCm. Для этого выполните следующие команды:

pip install torch torchvision torchaudio --extra-index-url https://rocm.github.io/whl/rocm6.1.html

Заключение

Теперь ваш компьютер настроен для работы с ROCm 6.1, и вы можете использовать PyTorch 2.4 для обучения моделей машинного обучения. Если у вас возникли какие-либо проблемы или дополнительные вопросы, пожалуйста, не стесняйтесь обращаться за помощью на форумах или в сообществе разработчиков ROCm.

Обязательно проверяйте наличие новых версий и выпусков для улучшения работы вашей системы. Удачи в ваших исследованиях и проектах в области машинного обучения!