AMD/ROCm: update base image string (#4137)

Co-authored-by: wunhuang <wunhuang@amd.com>
Co-authored-by: yichiche <yichiche@amd.com>
This commit is contained in:
kk
2025-03-06 19:38:54 +08:00
committed by GitHub
parent 98c73d71cb
commit b16af90bc3

View File

@@ -2,7 +2,7 @@
# docker build --build-arg SGL_BRANCH=v0.4.3.post3 -t v0.4.3.post3-rocm630 -f Dockerfile.rocm .
# default base image
ARG BASE_IMAGE="rocm/vllm-dev:20250114"
ARG BASE_IMAGE="rocm/sgl-dev:vllm20250114"
FROM $BASE_IMAGE AS base
USER root