Update Dockerfile

Dockerfile

@@ -1,6 +1,12 @@
 FROM pytorch/pytorch:2.7.1-cuda12.6-cudnn9-devel
 ENV CUDA_HOME=/usr/local/cuda
+ENV PATH="${CUDA_HOME}/bin:${PATH}"
+ENV LD_LIBRARY_PATH="${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}"
+ENV TORCH_CUDA_ARCH_LIST="7.5"
+ENV USE_CUDA=1
 
+# Verify CUDA setup
+RUN nvcc --version && which nvcc
 RUN apt-get update && apt-get install -y \
     git fish tmux curl ffmpeg \
     libgl1-mesa-glx libglib2.0-0 \
@@ -14,10 +20,7 @@ RUN useradd -m -u 1000 user
 ENV HF_HOME=/home/user/.cache/huggingface
 RUN mkdir -p $HF_HOME && chown -R user:user /home/user
 
-# Thread settings
-ARG USE_CUDA=1
-ARG TORCH_ARCH="7.0;7.5;8.0;8.6+PTX"
-ENV TORCH_CUDA_ARCH_LIST="${TORCH_ARCH}"
+
 ENV OMP_NUM_THREADS=1
 ENV MKL_NUM_THREADS=1
 ENV NUMEXPR_NUM_THREADS=1
@@ -46,6 +49,9 @@ RUN pip install --no-cache-dir -e ./video-sam2 && \
     pip install --no-cache-dir -e ./laser && \
     pip install --no-cache-dir -e ./vine_hf
 
+RUN cd groundingdino && \
+    python3 setup.py build_ext --force --inplace
+
 # Copy app
 COPY --chown=user:user . .