chore: bump v0.4.7 (#7038)
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user