If you are able to squeeze something usefull out of it - let me know by now i cant beat rocm/vllm-dev:nightly_main_20260211 for serving https://huggingface.co/truespirit/Ornith-1.5-397B-MXFP4
Unofficial community-built vLLM container for AMD Instinct™ MI355X (gfx950), based on the AMD ROCm™ PyTorch container.
This image is maintained by a private individual for experimentation and community use. It is not an official AMD, PyTorch, vLLM, FlashAttention, AITER, or Quark image and is not affiliated with, sponsored by, or endorsed by those projects or their respective maintainers.
docker pull truespirit/vllm-rocm10-gfx950:latest
| Component | Version / configuration |
|---|---|
| Base image | rocm/pytorch:rocm10.0_ubuntu24.04_py3.12_pytorch_release_2.13.0 |
| Python | 3.12 |
| PyTorch | 2.13.0 + ROCm 10 |
| vLLM | 0.27.0 |
| FlashAttention | 2.8.3 |
| AMD AITER | 0.1.20.post1 |
| AMD Quark | 0.12.post1 |
| Target GPU architecture | gfx950 |
| Primary target | AMD Instinct MI355X |
The image is approximately 57 GB.
This image exists to provide a reproducible vLLM userspace for gfx950, built around the ROCm 10 PyTorch base image.
Several components are built from source specifically for the target architecture.
The image is intentionally specialized for MI355X / gfx950. It should not be considered a general-purpose ROCm or vLLM image.
No model weights are included.
The host must have a working ROCm-compatible AMD GPU driver and expose the required GPU devices to Docker.
Example interactive start:
docker run --rm -it \
--device=/dev/kfd \
--device=/dev/dri \
--group-add video \
--ipc=host \
--shm-size 8G \
truespirit/vllm-rocm10-gfx950:latest
The default command is:
/bin/bash
On an MI355X system, run the included sanity check first:
verify-on-mi355x
For an initial vLLM test, using eager execution is recommended:
vllm serve <model> --enforce-eager
Some AITER/FlyDSL kernels are compiled by JIT during the first GPU warmup, so the first run can take longer than later runs.
The image uses:
HF_HOME=/models/hf
A host cache can for example be mounted there:
docker run --rm -it \
--device=/dev/kfd \
--device=/dev/dri \
--group-add video \
--ipc=host \
--shm-size 8G \
-v "$HOME/.cache/huggingface:/models/hf" \
truespirit/vllm-rocm10-gfx950:latest
Model files downloaded or mounted by the user are not part of this image and remain subject to their respective model licenses and terms.
gfx950.This container contains and redistributes third-party open-source software.
Nothing in this Docker Hub repository relicenses those components. Each component remains subject to its own copyright notices, license terms, attribution requirements, and other applicable conditions.
Major components include, among others:
| Project | Upstream license, best-effort summary |
|---|---|
| vLLM | Apache License 2.0 |
| FlashAttention | BSD 3-Clause |
| AMD AITER | MIT |
| AMD Quark | MIT |
| PyTorch | BSD-style project license and additional third-party licenses |
| AMD ROCm software | Multiple components with their respective licenses |
| Ubuntu packages | Individual package licenses |
| Python dependencies | Individual package licenses |
This table is only a convenience summary. The container contains additional direct and transitive dependencies that may use different licenses or include additional notices.
Users and redistributors should review the license and notice files supplied by the respective upstream projects and by the software installed inside the image.
Where an upstream license requires preservation of copyright notices, license text, attribution, NOTICE files, or similar information when redistributing binary software, those requirements continue to apply.
Names such as AMD, AMD Instinct, AMD ROCm, PyTorch, vLLM, FlashAttention, AITER, Quark, Docker, Hugging Face, and other product or project names are used only to identify compatibility and included software.
This project does not claim ownership of third-party trademarks, project names, software, or logos and does not imply endorsement by their owners.
AMD ROCm™ and AMD Instinct™ are trademarks of Advanced Micro Devices, Inc.
Other product names and trademarks belong to their respective owners.
This image is provided on an “AS IS” and “AS AVAILABLE” basis, without warranties or guarantees of any kind.
Use of this image is at your own risk. The maintainer makes no guarantee regarding fitness for a particular purpose, availability, performance, security, correctness, compatibility, or absence of defects.
This image may contain experimental software and very large compiled binaries. Review the image and its dependencies before using it in security-sensitive, production, regulated, or otherwise critical environments.
The maintainer is not responsible for model licenses, datasets, prompts, generated output, or other content that users choose to process with the image.
The fact that this image is published by a private individual or used for personal, research, or noncommercial purposes does not replace or override the licenses of the software contained in the image.
Before redistributing or modifying the image, users should independently verify that their intended use complies with all applicable upstream licenses and terms.
This README provides a best-effort description for informational purposes and is not legal advice.
Content type
Image
Digest
sha256:d99bc8d0d…
Size
20.4 GB
Last updated
28 days ago
docker pull truespirit/vllm-rocm10-gfx950