Запустил Pixal3D / TRELLIS.2 / hunyuan3D на AMD RX 9070 XT в Windows — без WSL и NVIDIA
Запустил Pixal3D / TRELLIS.2 на AMD RX 9070 XT в Windows — без WSL и NVIDIA
Привет, Пикабу!
Если у вас видеокарта AMD и вы когда-нибудь пытались запускать на ней нейросетевые 3D-инструменты, то наверняка видели этот бесконечный круг:
«Тут нужна CUDA».
«Поставь Linux».
«Попробуй WSL».
«Просто пересобери расширение».
«У нас вообще-то проверялось только на NVIDIA».
Мне это надоело, поэтому я решил проверить, можно ли нормально запустить Pixal3D / TRELLIS.2 на AMD прямо в Windows. Не через виртуалку, не через WSL и не с надеждой, что какая-нибудь CUDA-совместимость сама всё починит.
Коротко: получилось, но с важными оговорками. Я собрал установщик, 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...
Зачем я это выкладываю
Потому что одно дело — показать картинку и написать «у меня запустилось», и совсем другое — оставить нормальную отправную точку для следующего человека.
В репозитории лежат установщик, исходники изменений, колёса, контрольные суммы, тесты, результаты проверок и пример модели. Лицензии сторонних компонентов тоже сохранены.





























