Update Dockerfile

Dockerfile

@@ -1,68 +1,118 @@
-# Use the pre-built PyTorch Docker image with CUDA
-FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-devel
+# # Use the pre-built PyTorch Docker image with CUDA
+# FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-devel
 
-# Install build deps
-RUN apt-get update -y && apt-get install -y wget git vim ninja-build \
-    && apt-get clean && rm -f /var/lib/apt/lists/*_*
+# # Install build deps
+# RUN apt-get update -y && apt-get install -y wget git vim ninja-build \
+#     && apt-get clean && rm -f /var/lib/apt/lists/*_*
 
-# Set up environment variables
-ENV CUDA_HOME=/usr/local/cuda
-# ENV CONDA_DIR=/opt/conda
-# ENV PATH=$CONDA_DIR/bin:$PATH
+# # Set up environment variables
+# ENV CUDA_HOME=/usr/local/cuda
+# # ENV CONDA_DIR=/opt/conda
+# # ENV PATH=$CONDA_DIR/bin:$PATH
 
-# Set CUDA architecture flags
-ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
+# # Set CUDA architecture flags
+# ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
 
-WORKDIR /app
+# WORKDIR /app
 
-# Clone the InstantSplat repository and download the pre-trained model
-RUN git clone --recursive https://github.com/NVlabs/InstantSplat.git . && \
-    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/
+# # Clone the InstantSplat repository and download the pre-trained model
+# RUN git clone --recursive https://github.com/NVlabs/InstantSplat.git . && \
+#     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 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
-
-# RUN conda create -n instantsplat python=3.11 cmake=3.14.0 -y && \
-#     echo "source activate instantsplat" > ~/.bashrc && \
-#     /bin/bash -c "source activate instantsplat" && \
-#     conda install pytorch torchvision pytorch-cuda=11.8 -c pytorch -c nvidia -y && \
-#     cd InstantSplat && \
-#     pip install -r requirements.txt && \
-#     pip install submodules/simple-knn && \
+# 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
+#     pip install submodules/diff-gaussian-rasterization && \
+#     cd submodules/dust3r/croco/models/curope/ && \
+#     python setup.py build_ext --inplace
 
-RUN pip install -r requirements.txt
-RUN pip install imageio[ffmpeg]
+# # Create the conda environment and install dependencies
+# # RUN conda update -n base -c defaults conda
 
-# Set the working directory
-RUN useradd -m -u 1000 user
-RUN chown -R user:user /app
-USER user
+# # RUN conda create -n instantsplat python=3.11 cmake=3.14.0 -y && \
+# #     echo "source activate instantsplat" > ~/.bashrc && \
+# #     /bin/bash -c "source activate instantsplat" && \
+# #     conda install pytorch torchvision pytorch-cuda=11.8 -c pytorch -c nvidia -y && \
+# #     cd InstantSplat && \
+# #     pip install -r requirements.txt && \
+# #     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
+
+# RUN pip install -r requirements.txt
+# RUN pip install imageio[ffmpeg]
+
+# # Set the working directory
+# RUN useradd -m -u 1000 user
+# RUN chown -R user:user /app
+# USER user
+
+# COPY --chown=user . /app
+# # COPY --chown=user ./requirements.txt requirements.txt
+# # COPY ./requirements.txt requirements.txt
 
-COPY --chown=user . /app
-# COPY --chown=user ./requirements.txt requirements.txt
-# COPY ./requirements.txt requirements.txt
+# ENV PATH="/home/user/.local/bin:$PATH"
 
-ENV PATH="/home/user/.local/bin:$PATH"
+# # COPY --chown=user ./run_train_infer.sh run_train_infer.sh
+# # COPY ./run_train_infer.sh run_train_infer.sh
+# RUN chmod +x run_train_infer.sh
+
+
+# # Default command to run when the container starts
+# # CMD ["/bin/bash", "run_train_infer.sh"]
+
+# # Copy the current directory contents into the container at $HOME/app setting the owner to the user
+
+# CMD ["python", "app.py"]
+
+
+FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-devel
 
-# COPY --chown=user ./run_train_infer.sh run_train_infer.sh
-# COPY ./run_train_infer.sh run_train_infer.sh
-RUN chmod +x run_train_infer.sh
+ARG DEBIAN_FRONTEND=noninteractive
 
+ARG FORCE_CUDA=1
+ENV FORCE_CUDA=${FORCE_CUDA}
+
+ENV NVIDIA_VISIBLE_DEVICES \
+    ${NVIDIA_VISIBLE_DEVICES:-all}
+ENV NVIDIA_DRIVER_CAPABILITIES \
+    ${NVIDIA_DRIVER_CAPABILITIES:+$NVIDIA_DRIVER_CAPABILITIES,}graphics
+    
+RUN apt-get update -y && apt-get install -y git wget
+
+WORKDIR /
+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/
+
+WORKDIR /InstantSplat
+
+RUN pip3 install -r requirements.txt
+
+ARG TORCH_CUDA_ARCH_LIST="3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX"
+
+RUN pip3 install submodules/simple-knn
+RUN sed -i 's/p_view.z <= 0.2f/p_view.z <= 0.001f/g' /InstantSplat/submodules/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h
+RUN pip3 install submodules/diff-gaussian-rasterization
+
+RUN cd submodules/dust3r/croco/models/curope/ &&\
+    python3 setup.py build_ext --inplace
+    
+RUN pip3 install plyfile
+RUN apt-get install -y libgl1-mesa-dev libglib2.0-0
+
+ENV CUDA_VISIBLE_DEVICES=0
+
+RUN useradd -m -u 1000 user
+RUN chown -R user:user /InstantSplat
+USER user
 
-# Default command to run when the container starts
-# CMD ["/bin/bash", "run_train_infer.sh"]
+COPY --chown=user . /InstantSplat
 
-# Copy the current directory contents into the container at $HOME/app setting the owner to the user
 
 CMD ["python", "app.py"]