Add 'uv' usage to Dockerfile to speed up builds.

Author: puneetmatharu

ML-Frameworks/pytorch-aarch64/Dockerfile

@@ -39,24 +39,27 @@ WORKDIR /home/$DOCKER_USER
 RUN python -m venv /home/$DOCKER_USER/.venv
 ENV PATH="/home/$DOCKER_USER/.venv/bin:${PATH}"
 
+# Install uv for quicker package installations
+RUN pip install uv
+
 # Update to newer pip/setuptools/wheel (setuptools >= 70.0.0 due to CVE-2024-6345
 # and CVE-2025-47273, wheel >= 0.38.0 due to CVE-2022-40898) and delete old system
 # version (we essentially use apt:python3-pip to bootstrap pip)
-RUN pip install --upgrade pip~=25.2 setuptools~=78.1.1 wheel~=0.45.1
+RUN uv pip install --upgrade pip~=25.2 setuptools~=78.1.1 wheel~=0.45.1
 
 # Install non-torch requirements
 COPY --chown=$DOCKER_USER:$DOCKER_USER requirements.txt .
-RUN pip install -r requirements.txt
+RUN uv pip install -r requirements.txt
 
 # Bring wheels into image
 RUN test -n "${TORCH_WHEEL}" && test -n "${TORCH_AO_WHEEL}"
 COPY --chown=$DOCKER_USER:$DOCKER_USER ${TORCH_WHEEL} /home/$DOCKER_USER/
 COPY --chown=$DOCKER_USER:$DOCKER_USER ${TORCH_AO_WHEEL} /home/$DOCKER_USER/
 
 # Install wheels
-RUN set -eux && pip install --no-deps "$(basename "$TORCH_WHEEL")" && rm "$(basename "$TORCH_WHEEL")"
-RUN set -eux && pip install --no-deps "$(basename "$TORCH_AO_WHEEL")" && rm "$(basename "$TORCH_AO_WHEEL")"
-RUN pip install --pre torchvision==0.25.0.dev20260130 --index-url https://download.pytorch.org/whl/nightly/cpu --no-deps
+RUN set -eux && uv pip install --no-deps "$(basename "$TORCH_WHEEL")" && rm "$(basename "$TORCH_WHEEL")"
+RUN set -eux && uv pip install --no-deps "$(basename "$TORCH_AO_WHEEL")" && rm "$(basename "$TORCH_AO_WHEEL")"
+RUN uv pip install --pre torchvision==0.25.0.dev20260130 --index-url https://download.pytorch.org/whl/nightly/cpu --no-deps
 
 # Copy examples/tests into image
 COPY --chown=$DOCKER_USER:$DOCKER_USER examples/ /home/$DOCKER_USER/
@@ -74,7 +77,6 @@ ENV DOCKER_USER=${USERNAME}
 # Runtime OS bits + UI
 RUN set -eux && \
     apt-get update && \
-    apt-get install -y --no-install-recommends && \
     rm -rf /var/lib/apt/lists/* && \
     if ! id "$DOCKER_USER" >/dev/null 2>&1; then useradd --create-home -s /bin/bash "$DOCKER_USER"; fi && \
     echo '[ -n "$TERM" -a -r "$HOME/welcome.txt" ] && cat "$HOME/welcome.txt"' >> /etc/bash.bashrc && \