Update Dockerfile

Dockerfile

@@ -20,6 +20,9 @@ RUN git clone --recursive https://github.com/NVlabs/InstantSplat.git && \
 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"
+
 # Create the conda environment and install dependencies
 RUN conda update -n base -c defaults conda
 
@@ -41,7 +44,8 @@ RUN conda create -n instantsplat python=3.11 cmake=3.14.0 -y && \
 #     pip install submodules/diff-gaussian-rasterization
 
 # Compile the CUDA kernels for RoPE
-RUN cd InstantSplat/submodules/dust3r/croco/models/curope/ && \
+RUN /bin/bash -c "source activate instantsplat" && \
+    cd InstantSplat/submodules/dust3r/croco/models/curope/ && \
     python setup.py build_ext --inplace
 
 RUN useradd -m -u 1000 user