Update Dockerfile

Dockerfile

@@ -22,7 +22,28 @@ ENV VLLM_USE_TRITON_FLASH_ATTN=0
 # Choose XFORMERS that most stable for T4
 ENV VLLM_ATTENTION_BACKEND=XFORMERS
 
-RUN apt-get update && apt-get install -y python3 python3-pip git
+ARG PYTHON_VERSION=3.12
+
+# The PyPA get-pip.py script is a self contained script+zip file, that provides
+# both the installer script and the pip base85-encoded zip archive. This allows
+# bootstrapping pip in environment where a dsitribution package does not exist.
+#
+# By parameterizing the URL for get-pip.py installation script, we allow
+# third-party to use their own copy of the script stored in a private mirror.
+# We set the default value to the PyPA owned get-pip.py script.
+#
+# Reference: https://pip.pypa.io/en/stable/installation/#get-pip-py
+ARG GET_PIP_URL="https://bootstrap.pypa.io/get-pip.py"
+
+RUN apt-get update -y \
+    && apt-get install -y git python${PYTHON_VERSION} python${PYTHON_VERSION}-dev python${PYTHON_VERSION}-venv \
+    && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 1 \
+    && update-alternatives --set python3 /usr/bin/python${PYTHON_VERSION} \
+    && ln -sf /usr/bin/python${PYTHON_VERSION}-config /usr/bin/python3-config \
+    && curl -sS ${GET_PIP_URL} | python${PYTHON_VERSION} \
+    && python3 --version && python3 -m pip --version
+    
+# RUN apt-get update && apt-get install -y python3 python3-pip git
 RUN pip3 install --upgrade pip
 
 # Install vLLM