Skip to main content

GPUStack Runner is library for registering runnable accelerated backends and services in GPUStack.

Project description

GPUStack Runner

This repository serves as the Docker image pack center for GPUStack Runner. It provides a collection of Dockerfiles to build images for various inference services across different accelerated backends.

Agenda

Onboard Services

[!TIP]

  • The list below shows the accelerated backends and inference services available in the latest release. For support of backends or services not shown here, please refer to previous release tags.
  • Deprecated inference service versions in the latest release are marked with strikethrough formatting. They may still be available in previous releases, and not recommended for new deployments.
  • Polished inference service versions in the latest release are marked with bold formatting. If they are using in your deployment, it is recommended to pull the latest images and upgrade.

The following table lists the supported accelerated backends and their corresponding inference services with versions.

Ascend CANN

CANN Version
(Variant)
MindIE vLLM SGLang
8.5 (A3/910C) 2.3.0 0.18.0, 0.17.0(rc),
0.16.0(rc), 0.15.0(rc),
0.14.1(rc), 0.13.0
0.5.9, 0.5.8.post1
8.5 (910B) 2.3.0 0.18.0, 0.17.0(rc),
0.16.0(rc), 0.15.0(rc),
0.14.1(rc), 0.13.0
0.5.9, 0.5.8.post1
8.5 (310P) 2.3.0 0.18.0, 0.17.0(rc),
0.16.0(rc), 0.15.0(rc),
0.14.1(rc)
8.3 (A3/910C) 2.2.rc1 0.12.0(rc), 0.11.0 0.5.7, 0.5.6.post2
8.3 (910B) 2.2.rc1 0.12.0(rc), 0.11.0 0.5.7, 0.5.6.post2
8.3 (310P) 2.2.rc1
8.2 (A3/910C) 2.1.rc2 0.10.2(rc)
8.2 (910B) 2.1.rc2 0.10.2(rc), 0.10.0(rc),
0.9.1
8.2 (310P) 2.1.rc2 0.10.0(rc), 0.9.1

Iluvatar CoreX

CoreX Version
(Variant)
vLLM
4.2 0.8.3

NVIDIA CUDA

[!NOTE]

  • CUDA 13.0 supports Compute Capabilities: 7.5 8.0+PTX 8.9 9.0 10.0 10.3 12.0+PTX.
  • CUDA 12.9 supports Compute Capabilities: 7.5 8.0+PTX 8.9 9.0 10.0 10.3 12.0 12.1+PTX.
  • CUDA 12.8 supports Compute Capabilities: 7.5 8.0+PTX 8.9 9.0 10.0+PTX 12.0+PTX.
  • CUDA 12.6/12.4 supports Compute Capabilities: 7.5 8.0+PTX 8.9 9.0+PTX.
CUDA Version
(Variant)
vLLM SGLang VoxBox
13.0 0.20.1, 0.19.1,
0.18.1
12.9 0.20.1, 0.19.1,
0.18.1, 0.17.1,
0.16.0, 0.15.1,
0.14.1, 0.13.0,
0.12.0, 0.11.2
0.5.9, 0.5.8.post1,
0.5.7, 0.5.6.post2
12.8 0.17.1, 0.16.0,
0.15.1, 0.14.1,
0.13.0, 0.12.0,
0.11.2, 0.10.2
0.5.9, 0.5.8.post1,
0.5.7, 0.5.6.post2,
0.5.5.post3
0.0.21
12.6 0.15.1, 0.14.1,
0.13.0, 0.12.0,
0.11.2, 0.10.2
0.0.21

Hygon DTK

DTK Version
(Variant)
vLLM
25.04 0.11.0, 0.9.2, 0.8.5

T-Head HGGC

HGGC Version
(Variant)
vLLM SGLang
12.3 0.12.0, 0.11.1 0.5.6, 0.5.5

MetaX MACA

MACA Version
(Variant)
vLLM SGLang
3.3 0.11.2 0.5.6
3.2 0.10.2
3.0 0.9.1

MThreads MUSA

MUSA Version
(Variant)
vLLM SGLang
4.3.2 0.5.7
4.1.0 0.9.2

AMD ROCm

[!NOTE]

  • ROCm 7.1/7.0 supports LLVM targets: gfx908 gfx90a gfx942 gfx950 gfx1030 gfx1100 gfx1101 gfx1200 gfx1201 gfx1150 gfx1151.
  • ROCm 6.4 supports LLVM targets: gfx908 gfx90a gfx942 gfx1030 gfx1100.

[!WARNING]

  • ROCm 7.0 vLLM 0.11.2 are reusing the official ROCm 6.4 PyTorch 2.9 wheel package rather than a ROCm 7.0 specific PyTorch build. Although supports ROCm 7.0 in vLLM 0.11.2, gfx1150/gfx1151 are not supported yet.
  • ROCm 6.4 vLLM 0.13.0 supports gfx903 gfx90a gfx942 only.
  • ROCm 6.4 SGLang supports gfx942 only.
  • ROCm 7.0 SGLang supports gfx950 only.
ROCm Version
(Variant)
vLLM SGLang
7.2 0.20.1, 0.19.1
7.1 0.17.1
7.0 0.18.1,
0.16.0, 0.15.1,
0.14.1, 0.13.0,
0.12.0, 0.11.2
0.5.9, 0.5.8.post1,
0.5.7, 0.5.6.post2
6.4 0.16.0,
0.15.1, 0.14.1,
0.13.0, 0.12.0,
0.11.2, 0.10.2
0.5.8.post1, 0.5.7,
0.5.6.post2, 0.5.5.post3

Directory Structure

The pack skeleton is organized by backend:

pack
├── {BACKEND 1}
│   └── Dockerfile
├── {BACKEND 2}
│   └── Dockerfile
├── {BACKEND 3}
│   └── Dockerfile
├── ...
│   └── Dockerfile
└── {BACKEND N}
    └── Dockerfile

Dockerfile Convention

Each Dockerfile follows these conventions:

  • Begin with comments describing the package logic in steps and usage of build arguments (ARGs).
  • Use ARG for all required and optional build arguments. If a required argument is unused, mark it as (PLACEHOLDER).
  • Use heredoc syntax for RUN commands to improve readability.

Example Dockerfile Structure

# Describe package logic and ARG usage.
#
ARG PYTHON_VERSION=...                                 # REQUIRED
ARG CMAKE_MAX_JOBS=...                                 # REQUIRED
ARG {OTHERS}                                           # OPTIONAL
ARG {BACKEND}_VERSION=...                              # REQUIRED
ARG {BACKEND}_VERSION_EXTRA=...                        # OPTIONAL
ARG {BACKEND}_ARCHS=...                                # REQUIRED
ARG {BACKEND}_{OTHERS}=...                             # OPTIONAL
ARG {SERVICE}_BASE_IMAGE=...                           # REQUIRED
ARG {SERVICE}_VERSION=...                              # REQUIRED
ARG {SERVICE}_{OTHERS}=...                             # OPTIONAL
ARG {SERVICE}_{FRAMEWORK}_VERSION=...                  # REQUIRED
ARG {SERVICE}_{FRAMEWORK}_{OTHERS}=...                 # OPTIONAL

# Stage Bake Runtime
FROM {BACKEND DEVEL IMAGE} AS runtime
SHELL ["/bin/bash", "-eo", "pipefail", "-c"]
ARG TARGETPLATFORM
ARG TARGETOS
ARG TARGETARCH
ARG ...
RUN <<EOF
    # TODO: install runtime dependencies
EOF

# Stage Install Service
FROM {BACKEND}_BASE_IMAGE AS {service}
SHELL ["/bin/bash", "-eo", "pipefail", "-c"]
ARG TARGETPLATFORM
ARG TARGETOS
ARG TARGETARCH
ARG ...
RUN <<EOF
    # TODO: install service and dependencies
EOF

WORKDIR /
ENTRYPOINT [ "tini", "--" ]

Docker Image Naming Convention

The Docker image naming convention is as follows:

  • Multi-architecture image names: {NAMESPACE}/{REPOSITORY}:{TAG}.
  • Single-architecture image tags: {BACKEND}{BACKEND_VERSION%.*}[-{BACKEND_VARIANT}]-{SERVICE}{SERVICE_VERSION}-{OS}-{ARCH}.
  • Multi-architecture image tags: {BACKEND}{BACKEND_VERSION%.*}[-{BACKEND_VARIANT}]-{SERVICE}{SERVICE_VERSION}[-dev].
  • All names adn tags must be lowercase.

Example

  • NAMESPACE: gpustack
  • REPOSITORY: runner
Accelerated Backend OS/ARCH Inference Service Single-Arch Image Name Multi-Arch Image Name
Ascend CANN 910b linux/amd64 vLLM gpustack/runner:cann8.1-910b-vllm0.9.2-linux-amd64 gpustack/runner:cann8.1-910b-vllm0.9.2
Ascend CANN 910b linux/arm64 vLLM gpustack/runner:cann8.1-910b-vllm0.9.2-linux-arm64 gpustack/runner:cann8.1-910b-vllm0.9.2
NVIDIA CUDA 12.8 linux/amd64 vLLM gpustack/runner:cuda12.8-910b-vllm0.9.2-linux-amd64 gpustack/runner:cuda12.8-910b-vllm0.9.2
NVIDIA CUDA 12.8 linux/arm64 vLLM gpustack/runner:cuda12.8-910b-vllm0.9.2-linux-arm64 gpustack/runner:cuda12.8-910b-vllm0.9.2

Build and Release Workflow

  1. Build single architecture images for OS/ARCH, e.g. gpustack/runner:cann8.1-910b-vllm0.9.2-linux-amd64.
  2. Combine single-architecture images into a multiple architectures image, e.g. gpustack/runner:cann8.1-910b-vllm0.9.2-dev.
  3. After testing, rename the multi-architecture image to the final tag, e.g. gpustack/runner:cann8.1-910b-vllm0.9.2.

Integration Process

Ingesting a New Accelerated Backend

To add support for a new accelerated backend:

  1. Create a new directory under pack/ named with the new backend.
  2. Add a Dockerfile in the new directory following the Dockerfile Convention.
  3. Update pack.yml, discard.yml and prune.yml to include the new backend in the build matrix.
  4. Update matrix.yml to include the new backend and its variants.
  5. Update _RE_DOCKER_IMAGE in runner.py to recognize the new backend.
  6. [Optional] Update tests if necessary.

Ingesting a New Inference Service

To add support for a new inference service:

  1. Modify the Dockerfile of the relevant backend in pack/{BACKEND}/Dockerfile to include the new service.
  2. Update pack.yml to include the new service in the build matrix.
  3. Update matrix.yml to include the new service.
  4. Update _RE_DOCKER_IMAGE in runner.py to recognize the new service.
  5. [Optional] Update tests if necessary.

License

Copyright (c) 2025 The GPUStack authors

Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at LICENSE file for details.

Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

gpustack_runner-0.1.26.post1.tar.gz (13.4 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

gpustack_runner-0.1.26.post1-py3-none-any.whl (29.2 kB view details)

Uploaded Python 3

File details

Details for the file gpustack_runner-0.1.26.post1.tar.gz.

File metadata

File hashes

Hashes for gpustack_runner-0.1.26.post1.tar.gz
Algorithm Hash digest
SHA256 32471e283c625fb31ae3257b9c9361e30c27cbf28c887a30f0886370546226ac
MD5 88bf8c9a94a5a166dc87d74fe2c64dcb
BLAKE2b-256 efad4a61afd6ba04ae78d03481f1b156b6aa412a94b99c884c565e799439e3d4

See more details on using hashes here.

File details

Details for the file gpustack_runner-0.1.26.post1-py3-none-any.whl.

File metadata

File hashes

Hashes for gpustack_runner-0.1.26.post1-py3-none-any.whl
Algorithm Hash digest
SHA256 80828afb7efae4f62278f621995c50f5e60b541c97dfb382a498f9407218721d
MD5 c311d9841fefea4b1c2a4b826a0c4789
BLAKE2b-256 587ed4ecf3535e4c56d5907b11243330aad5c04b1ec8b9b079ae4fafe2f1ca2a

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page