Update Dockerfile

Dockerfile

@@ -1,10 +1,6 @@
 # Use the pre-built PyTorch Docker image with CUDA
 FROM pytorch/pytorch:2.1.2-cuda11.8-cudnn8-devel
 
-# Set up environment variables
-# ENV CONDA_DIR=/opt/conda
-# ENV PATH=$CONDA_DIR/bin:$PATH
-
 # Install build deps
 RUN apt-get update -y && apt-get install -y wget git vim \
     && apt-get clean && rm -f /var/lib/apt/lists/*_*
@@ -19,9 +15,13 @@ RUN git clone --recursive https://github.com/NVlabs/InstantSplat.git && \
     # wget https://download.europe.naverlabs.com/ComputerVision/DUSt3R/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth -P checkpoints/ && \
     # cd ../../
 
+# Set up environment variables
+ENV CONDA_DIR=/opt/conda
+ENV PATH=$CONDA_DIR/bin:$PATH
+
 # Create the conda environment and install dependencies
-RUN conda create -n instantsplat python=3.11 cmake=3.14.0 && \
-    conda activate instantsplat && \
+RUN conda create -n instantsplat python=3.11 cmake=3.14.0
+RUN conda activate instantsplat && \
     conda install pytorch torchvision pytorch-cuda=12.1 -c pytorch -c nvidia -y && \
     cd InstantSplat && \
     pip install -r requirements.txt && \