Devcontainer: Optimize apt-get commands to reduce Docker image size (#152882)

wdvr · pytorchmergebot · commit 5fe58ab5bd9e · 2025-05-06T20:33:02.000Z
## Summary - Added --no-install-recommends flag to all apt-get install commands to reduce unnecessary dependencies - Added apt-get clean after package installations to remove package cache and reduce image size - Combined multiple apt commands into single instructions to reduce Docker image layers ## Test plan Test by building the devcontainer and verifying functionality while ensuring reduced image size Pull Request resolved: #152882 Approved by: https://github.com/cyyever, https://github.com/atalman, https://github.com/Skylion007
diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
@@ -14,11 +14,8 @@ RUN apt-get -y update && \
     libopenblas-dev
 
 # Tools needed for llvm
-RUN apt-get install -y \
-    lsb-release \
-    wget \
-    software-properties-common \
-    gnupg
+RUN apt-get install --no-install-recommends -y lsb-release wget software-properties-common gnupg && \
+    sudo apt-get clean -y
 
 # Create Python virtual environment
 # RUN python3 -m venv /opt/venv
@@ -33,7 +30,8 @@ RUN if [ -n "$CLANG_VERSION" ]; then \
     ./llvm.sh "${CLANG_VERSION}"; \
     echo 'export CC=clang' >> ~/.bashrc; \
     echo 'export CXX=clang++' >> ~/.bashrc; \
-    apt install -y clang libomp-dev; \
+    apt-get install --no-install-recommends -y clang libomp-dev && \
+    apt-get clean -y; \
     fi
 
 
@@ -43,7 +41,8 @@ RUN if [ -n "$CUDA_VERSION" ]; then \
     CUDA_REPO_VERSION=$(echo ${CUDA_VERSION} | sed 's/\./\-/g'); \
     wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb && \
     dpkg -i cuda-keyring_1.0-1_all.deb && \
-    apt-get install -y cuda-toolkit-${CUDA_VERSION}; \
+    apt-get install --no-install-recommends -y cuda-toolkit-${CUDA_VERSION} && \
+    apt-get clean -y; \
     fi
 
 # Set PATH for CUDA
