Skip to content

Commit 24b436e

Browse files
committed
Revert sglang changes as the python version is 3.10
Signed-off-by: pvijayakrish <[email protected]>
1 parent ccbb378 commit 24b436e

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

container/Dockerfile.sglang

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,7 @@ RUN apt-get update \
8585
# Python (using other python versions as needed)
8686
python${PYTHON_VERSION}-dev \
8787
python${PYTHON_VERSION}-venv \
88+
python${PYTHON_VERSION}-distutils \
8889
python3-pip \
8990
# Build essentials
9091
build-essential \
@@ -156,8 +157,6 @@ RUN apt-get update \
156157
&& update-alternatives --set python3 /usr/bin/python${PYTHON_VERSION} \
157158
&& update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 1 \
158159
&& update-alternatives --set python /usr/bin/python${PYTHON_VERSION} \
159-
# Remove EXTERNALLY-MANAGED marker to allow pip installs in Docker (PEP 668)
160-
&& rm -f /usr/lib/python${PYTHON_VERSION}/EXTERNALLY-MANAGED \
161160
# Set up locale
162161
&& locale-gen en_US.UTF-8 \
163162
# Cleanup
@@ -253,7 +252,7 @@ RUN --mount=type=secret,id=aws-key-id,env=AWS_ACCESS_KEY_ID \
253252
cmake --build build --target install -j${CMAKE_BUILD_PARALLEL_LEVEL} && \
254253
/tmp/use-sccache.sh show-stats "NVSHMEM"
255254

256-
# Build nvshmem4py wheels separately for the configured Python version (CUDA 12) to avoid building the python library twice for multiple python versions
255+
# Build nvshmem4py wheels separately (Python 3.10, CUDA 12) to avoid building the python library twice for multiple python versions
257256
# Need to reconfigure with PYTHON_LIB=ON to add the nvshmem4py subdirectory
258257
RUN --mount=type=secret,id=aws-key-id,env=AWS_ACCESS_KEY_ID \
259258
--mount=type=secret,id=aws-secret-id,env=AWS_SECRET_ACCESS_KEY \
@@ -500,4 +499,4 @@ RUN python3 -m pip install --no-cache-dir \
500499
tabulate
501500

502501
ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]
503-
CMD []
502+
CMD []

0 commit comments

Comments
 (0)