Update Dockerfile

Dockerfile

@@ -1,7 +1,6 @@
 # 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 \
     && apt-get clean && rm -f /var/lib/apt/lists/*_*
@@ -26,23 +25,22 @@ ENV TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6"
 # 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 && \
-    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
-
-# # Create the conda environment and install dependencies
-# RUN cd InstantSplat && \
+# 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 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
+
 # Compile the CUDA kernels for RoPE
 RUN /bin/bash -c "source activate instantsplat" && \
     cd InstantSplat/submodules/dust3r/croco/models/curope/ && \