Update Dockerfile

Dockerfile

@@ -2,7 +2,7 @@
 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 \
+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
@@ -16,8 +16,9 @@ RUN git clone --recursive https://github.com/NVlabs/InstantSplat.git && \
     # cd ../../
 
 # Set up environment variables
-ENV CONDA_DIR=/opt/conda
-ENV PATH=$CONDA_DIR/bin:$PATH
+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"