diff --git a/Dockerfile b/Dockerfile
index 9e2723e..278d289 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -38,12 +38,11 @@ RUN apt-get update && apt-get install -y software-properties-common && \
     ninja-build \
     && rm -rf /var/lib/apt/lists/*
 
-# Set Python 3.12 as default
+# Set Python 3.12 as default and bootstrap pip
 RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.12 1 && \
-    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.12 1
-
-# Upgrade pip
-RUN python -m pip install --upgrade pip setuptools wheel
+    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.12 1 && \
+    python3.12 -m ensurepip --upgrade && \
+    python3.12 -m pip install --upgrade pip setuptools wheel
 
 # Install PyTorch 2.8.0+cu128 and triton 3.4.0
 RUN pip install \