diff --git a/docker/Dockerfile b/docker/Dockerfile index e7a878d31cd..aa9f3a4e4cf 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -25,15 +25,6 @@ WORKDIR /sgl-workspace ARG CUDA_VERSION RUN python3 -m pip install --upgrade pip setuptools wheel html5lib six \ - && if [ "$CUDA_VERSION" = "12.1.1" ]; then \ - python3 -m pip install torch --index-url https://download.pytorch.org/whl/cu121; \ - elif [ "$CUDA_VERSION" = "12.4.1" ]; then \ - python3 -m pip install torch --index-url https://download.pytorch.org/whl/cu124; \ - elif [ "$CUDA_VERSION" = "11.8.0" ]; then \ - python3 -m pip install torch --index-url https://download.pytorch.org/whl/cu118; \ - else \ - echo "Unsupported CUDA version: $CUDA_VERSION" && exit 1; \ - fi \ && git clone --depth=1 https://github.com/sgl-project/sglang.git \ && cd sglang \ && if [ "$BUILD_TYPE" = "srt" ]; then \