TsubasaR

Коротко: получилось, но с важными оговорками. Я собрал установщик, HIP-расширения, проверки и рабочий пример для Radeon RX 9070 XT. Всё выложил на GitHub.

Что именно работает

Проверенная конфигурация такая:

• Windows x64;

• AMD Radeon RX 9070 XT (gfx1201);

• ROCm 7.2.1;

• PyTorch 2.9.1+rocm7.2.1;

• CPython 3.12.

Для неё я собрал нативные Windows-колёса CuMesh, FlexGEMM и O-Voxel. CuMesh BVH заменён на cuBQL, для которого тоже есть GPU-проверки.

Установщик создаёт отдельный Python-runtime и не трогает системный Python. ROCm и PyTorch скачиваются с официального репозитория AMD, размер файлов и SHA-256 проверяются автоматически. Драйверы установщик не заменяет и не откатывает, перезагрузка не требуется.

После чистой установки прошли:

• 9 из 9 базовых проверок;

• 6 из 6 обязательных проверок HIP-расширений;

• 44 из 44 Python-теста в AMD-окружении;

• проверки sparse SDPA, NAF, FlexGEMM, O-Voxel, CuMesh и cuBQL BVH.

В репозитории есть пример модели Baton Standard 1024 с текстурой 2048 × 2048 и сохранённым результатом генерации.

А теперь без розовых соплей

Это не универсальная поддержка всех видеокарт AMD.

Я проверял только RX 9070 XT. Если у вас другая карта, я не буду кричать «всё поддерживается», пока кто-то реально не прогонит на ней весь комплект проверок.

Это также не готовый пакет со всеми нейросетевыми весами. Полные веса моделей в релиз не входят. Установка и зелёные проверки означают, что среда и расширения готовы, а не что после одного клика автоматически появятся все режимы генерации.

И да, это не волшебный переводчик любой CUDA-программы в AMD. Пришлось разбираться с конкретными зависимостями, собирать нужные расширения под HIP и проверять, что они действительно работают, а не просто импортируются.

Покупать RX 9070 XT только ради этой сборки я бы не советовал. А вот если такая карта уже есть — можно попробовать.

Свежий релиз v0.1.3

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

Изначально часть тестов случайно ссылалась на мой локальный каталог afamd. У меня всё проходило, а GitHub Actions честно отвечал:

«Файл не найден».

Классическое «у меня работает».

Я убрал эти пути, разделил автономные тесты и тесты, которым действительно нужна AMD-карта, настроил CI на Windows и добавил бейдж статуса в README.

Теперь GitHub после каждого изменения запускает 20 автономных проверок, которым не нужны ROCm, PyTorch и GPU. Последний запуск полностью зелёный. Полный AMD-набор по-прежнему проверяется отдельно на видеокарте.

То есть в v0.1.3 колёса и runtime не изменились, зато проект стало проще проверять и развивать дальше.

1. Батон — пример Pixal3D на AMD

«Батон, созданный в Pixal3D / TRELLIS.2 на AMD RX 9070 XT. Четыре ракурса модели, текстура 2048 × 2048».

2. Земля — дневная сторона

«Отдельный эксперимент с игровым рендером Земли: облака без видеотекстуры».

3. Земля — граница дня и ночи

«Граница дня и ночи: облака, освещение и огни городов на тёмной стороне».

Как попробовать

Скачайте полный ZIP из релиза и распакуйте его. Затем откройте PowerShell в корне проекта.

Сначала сухой запуск — без скачивания и изменений:

powershellКопировать

powershell -NoProfile -ExecutionPolicy Bypass -File installer/amd/Pixal3DNative.ps1 -RuntimeRoot C:\AMD-AI\Pixal3D-ROCm -DryRun

Если всё устраивает, запускайте установку и проверки:

powershellКопировать

powershell -NoProfile -ExecutionPolicy Bypass -File installer/amd/Pixal3DNative.ps1 -RuntimeRoot C:\AMD-AI\Pixal3D-ROCm

Каталог назначения лучше указать новый и не смешивать эту среду с существующими NVIDIA- или WSL-установками.

https://github.com/tsubasaprogame/Pixal3D-TRELLIS.2-AMD-ROCm...

Зачем я это выкладываю

Потому что одно дело — показать картинку и написать «у меня запустилось», и совсем другое — оставить нормальную отправную точку для следующего человека.

В репозитории лежат установщик, исходники изменений, колёса, контрольные суммы, тесты, результаты проверок и пример модели. Лицензии сторонних компонентов тоже сохранены.