Update Dockerfile

Dockerfile

@@ -5,15 +5,6 @@ FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-devel
 RUN apt-get update -y && apt-get install -y wget git vim ninja-build \
     && apt-get clean && rm -f /var/lib/apt/lists/*_*
 
-# Clone the InstantSplat repository and download the pre-trained model
-RUN git clone --recursive https://github.com/NVlabs/InstantSplat.git && \
-    cd InstantSplat && \
-    git submodule update --init --recursive && \
-    cd submodules/dust3r/ && \
-    mkdir -p checkpoints/ && \
-    wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/ && \
-    cd ../../
-
 # Set up environment variables
 ENV CUDA_HOME=/usr/local/cuda
 # ENV CONDA_DIR=/opt/conda
@@ -22,6 +13,23 @@ ENV CUDA_HOME=/usr/local/cuda
 # Set CUDA architecture flags
 ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
 
+# Clone the InstantSplat repository and download the pre-trained model
+RUN git clone --recursive https://github.com/NVlabs/InstantSplat.git app/ && \
+    cd app && \
+    git submodule update --init --recursive && \
+    cd submodules/dust3r/ && \
+    mkdir -p checkpoints/ && \
+    wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/
+    
+RUN useradd -m -u 1000 user
+WORKDIR /app
+
+RUN pip install submodules/simple-knn && \
+    sed -i '154s/p_view.z <= 0.2f/p_view.z <= 0.001f/' submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h && \
+    pip install submodules/diff-gaussian-rasterization && \
+    cd submodules/dust3r/croco/models/curope/ && \
+    python setup.py build_ext --inplace
+
 # Create the conda environment and install dependencies
 # RUN conda update -n base -c defaults conda
 
@@ -35,29 +43,17 @@ ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
 #     sed -i '154s/p_view.z <= 0.2f/p_view.z <= 0.001f/' submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h && \
 #     pip install submodules/diff-gaussian-rasterization
 
-COPY --chown=user requirements.txt InstantSplat/requirements.txt
-RUN cd InstantSplat && \
-    pip install -r requirements.txt
-
-RUN cd InstantSplat && \
-    pip install submodules/simple-knn && \
-    sed -i '154s/p_view.z <= 0.2f/p_view.z <= 0.001f/' submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h && \
-    pip install submodules/diff-gaussian-rasterization && \
-    cd submodules/dust3r/croco/models/curope/ && \
-    python setup.py build_ext --inplace
+# Set the working directory
 
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install -r requirements.txt
 
-RUN useradd -m -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
-COPY --chown=user run_train_infer.sh InstantSplat/run_train_infer.sh
-RUN chmod +x InstantSplat/run_train_infer.sh
-
-# Set the working directory
-WORKDIR InstantSplat/
+COPY --chown=user ./run_train_infer.sh run_train_infer.sh
+RUN chmod +x run_train_infer.sh
 
-RUN chown -R user:user /workspace/InstantSplat
 
 # Default command to run when the container starts
 CMD ["/bin/bash", "run_train_infer.sh"]