From 989e7df60f7c72e9ea6935239a0fb9d239a435cc Mon Sep 17 00:00:00 2001
From: Ying Sheng <sqy1415@gmail.com>
Date: Sat, 3 Aug 2024 02:53:31 +0000
Subject: [PATCH] test docker

---
 README.md         |  4 ++--
 docker/Dockerfile | 10 ++++++----
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 85f464fce93..a14dda3c234 100644
--- a/README.md
+++ b/README.md
@@ -54,8 +54,8 @@ pip install flashinfer -i https://flashinfer.ai/whl/cu121/torch2.3/
 
 ### Method 2: From source
 ```
-# Use the stable v0.2.9.post1 branch
-git clone -b v0.2.9.post1 https://github.com/sgl-project/sglang.git
+# Use the stable v0.2.9.post2 branch
+git clone -b v0.2.9.post2 https://github.com/sgl-project/sglang.git
 cd sglang
 
 pip install --upgrade pip
diff --git a/docker/Dockerfile b/docker/Dockerfile
index af7982f84a6..af8f34b5414 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,6 +1,6 @@
 ARG CUDA_VERSION=12.1.1
 
-FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04
+FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
 
 ARG PYTHON_VERSION=3
 
@@ -12,7 +12,7 @@ RUN echo 'tzdata tzdata/Areas select America' | debconf-set-selections \
     && apt-get install -y ccache software-properties-common \
     && add-apt-repository ppa:deadsnakes/ppa \
     && apt-get update -y \
-    && apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv python3-pip \
+    && apt-get install -y python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv python${PYTHON_VERSION}-pip \
     && if [ "${PYTHON_VERSION}" != "3" ]; then update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1; fi \
     && python3 --version \
     && python3 -m pip --version \
@@ -24,10 +24,12 @@ RUN apt-get update -y \
 
 WORKDIR /sgl-workspace
 
-RUN pip3 --no-cache-dir install --upgrade pip \
+RUN pip3 --no-cache-dir install --upgrade pip
+    && pip3 --no-cache-dir install --upgrade setuptools wheel \
+    && python3 -m pip install --upgrade --user pip \
     && git clone --depth=1 https://github.com/sgl-project/sglang.git \
     && cd sglang \
-    && pip3 --no-cache-dir install -e "python[all]" \
+    && pip --no-cache-dir install -e "python[all]" \
     && pip3 --no-cache-dir install flashinfer -i https://flashinfer.ai/whl/cu121/torch2.3/
 
 ENV DEBIAN_FRONTEND=interactive