chore: bump v0.4.7 (#7038)

This commit is contained in:
Yineng Zhang
2025-06-10 01:56:20 -07:00
committed by GitHub
parent fcde67b016
commit 4f723edd3b
8 changed files with 19 additions and 15 deletions

View File

@@ -37,9 +37,13 @@ RUN python3 -m pip install --upgrade pip setuptools wheel html5lib six \
else \
echo "Unsupported CUDA version: $CUDA_VERSION" && exit 1; \
fi \
&& python3 -m pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu${CUINDEX} \
&& if [ "$CUDA_VERSION" = "12.4.1" ]; then \
python3 -m pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu126; \
else \
python3 -m pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu${CUINDEX}; \
fi \
&& cd sglang \
&& python3 -m pip --no-cache-dir install -e "python[${BUILD_TYPE}]" --find-links https://flashinfer.ai/whl/cu${CUINDEX}/torch2.6/flashinfer-python \
&& python3 -m pip --no-cache-dir install -e "python[${BUILD_TYPE}]" \
&& if [ "$CUDA_VERSION" = "12.8.1" ]; then \
python3 -m pip install nvidia-nccl-cu12==2.26.2.post1 --force-reinstall --no-deps; \
fi

View File

@@ -1,5 +1,5 @@
# Usage (to build SGLang ROCm docker image):
# docker build --build-arg SGL_BRANCH=v0.4.6.post5 -t v0.4.6.post5-rocm630 -f Dockerfile.rocm .
# docker build --build-arg SGL_BRANCH=v0.4.7 -t v0.4.7-rocm630 -f Dockerfile.rocm .
# default base image
ARG BASE_IMAGE="rocm/sgl-dev:vllm20250114"