From 501e5e2284222c05280a0ec4f58cd55098396719 Mon Sep 17 00:00:00 2001
From: basebala <basebala@users.noreply.github.com>
Date: Wed, 8 Jul 2026 21:46:16 -0700
Subject: [PATCH] Enhance Dockerfile with package installations

Updated the Dockerfile to install additional Python packages and upgrade setuptools.
---
 Dockerfile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Dockerfile b/Dockerfile
index 0b6451a80..41e9f9064 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -99,6 +99,7 @@ COPY --from=source /tmp/nerfstudio/ /tmp/nerfstudio
 RUN export TORCH_CUDA_ARCH_LIST="$(echo "$CUDA_ARCHITECTURES" | tr ';' '\n' | awk '$0 > 70 {print substr($0,1,1)"."substr($0,2)}' | tr '\n' ' ' | sed 's/ $//')" && \
     export MAX_JOBS=4 && \
     pip install --no-cache-dir torch==2.1.2+cu118 torchvision==0.16.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 && \
+    pip install --upgrade setuptools && \
     GSPLAT_VERSION="$(sed -n 's/.*gsplat==\s*\([^," '"'"']*\).*/\1/p' /tmp/nerfstudio/pyproject.toml)" && \
     pip install --no-cache-dir git+https://github.com/nerfstudio-project/gsplat.git@v${GSPLAT_VERSION} && \
     pip install --no-cache-dir /tmp/nerfstudio 'numpy<2.0.0' && \