|
|
@@ -24,6 +24,7 @@ ARG PYTHON_VERSION=3.12
|
|
|
ARG REGISTRY_MIRROR=docker.1ms.run
|
|
|
ARG HIGRESS_REGISTRY=docker.io
|
|
|
ARG GPUSTACK_BASE_IMAGE=base
|
|
|
+ARG ROCM_REGISTRY=docker.io
|
|
|
ARG GPUSTACK_RUNTIME_ROCM_VERSION=7.0.2
|
|
|
ARG GPUSTACK_RUNTIME_DOCKER_MIRRORED_NAME_FILTER_LABELS
|
|
|
ARG HIGRESS_VERSION=2.1.9
|
|
|
@@ -304,7 +305,7 @@ ENV S6_KEEP_ENV=1 \
|
|
|
# Vendor ROCm libraries from ROCm base image,
|
|
|
# now only linux/amd64 is supported.
|
|
|
# Must build on linux/amd64 platform.
|
|
|
-FROM --platform=${BUILDPLATFORM} ${REGISTRY_MIRROR}/rocm/dev-ubuntu-22.04:${GPUSTACK_RUNTIME_ROCM_VERSION} AS rocm-base
|
|
|
+FROM --platform=${BUILDPLATFORM} ${ROCM_REGISTRY}/rocm/dev-ubuntu-22.04:${GPUSTACK_RUNTIME_ROCM_VERSION} AS rocm-base
|
|
|
|
|
|
FROM ${GPUSTACK_BASE_IMAGE} AS gpustack
|
|
|
SHELL ["/bin/bash", "-eo", "pipefail", "-c"]
|
