Revert "fix #2528"

This reverts commit b749db9.
sgl-project · Dec 21, 2024 · 0d076ad · 0d076ad
1 parent b749db9
commit 0d076ad
Showing 1 changed file with 0 additions and 9 deletions.
diff --git 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 \