Update Dockerfile

Dockerfile

@@ -1,5 +1,8 @@
-# Use the pre-built PyTorch Docker image with CUDA
-FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-devel
+
+RUN conda update -n base -c defaults conda
+ RUN# Use the pre-built PyTorch Docker image with
+
+FROM ch/pytorch:2.1.2-cuda11.8-cudnn8-devel
 
 # Install build deps
 RUN apt-get update -y && apt-get install -y wget git vim \
@@ -20,11 +23,14 @@ ENV CONDA_DIR=/opt/conda
 ENV PATH=$CONDA_DIR/bin:$PATH
 
 # 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=12.1 -c pytorch -c nvidia -y && \
-    cd InstantSplat && \
+    /bin/bash -c "source activate instantsplat"
+
+RUN conda install pytorch torchvision pytorch-cuda=12.1 -c pytorch -c nvidia -y
+
+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 && \