Update Dockerfile

Dockerfile

@@ -79,6 +79,10 @@ RUN cd /tmp && pip install -r requirements.txt
 RUN pip install https://download.is.tue.mpg.de/icon/HF/kaolin-0.11.0-cp38-cp38-linux_x86_64.whl
 RUN pip install https://download.is.tue.mpg.de/icon/HF/pytorch3d-0.7.0-cp38-cp38-linux_x86_64.whl
 RUN pip install pyembree
+
+# Verify CUDA installation
+RUN ls /usr/local/cuda/bin
+RUN nvcc --version
 RUN pip install git+https://github.com/YuliangXiu/neural_voxelization_layer.git
 
 ENV TRANSFORMERS_CACHE=/tmp
@@ -97,6 +101,4 @@ COPY --chown=user . $HOME/app
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
 
-RUN which nvcc
-
 CMD ["python", "app.py"]