docker

Dockerfile

@@ -0,0 +1,55 @@
+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 \
+    build-essential ninja-build python3.10-venv \
+    && rm -rf /var/lib/apt/lists/*
+    
+# Create non-root user
+RUN useradd -m -u 1000 user
+
+# Create HF cache with correct ownership
+ENV HF_HOME=/home/user/.cache/huggingface
+RUN mkdir -p $HF_HOME && chown -R user:user /home/user
+
+
+ENV OMP_NUM_THREADS=1
+ENV MKL_NUM_THREADS=1
+ENV NUMEXPR_NUM_THREADS=1
+
+# Switch to user before creating venv
+USER user
+WORKDIR /home/user/app
+
+# Create virtualenv
+RUN python3 -m venv /home/user/venv
+ENV PATH="/home/user/venv/bin:$PATH"
+
+# Dependencies
+COPY --chown=user:user requirements.txt .
+RUN pip install --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
+
+# Editable installs
+RUN pip install --no-cache-dir -e ./src/video-sam2 && \
+    pip install --no-cache-dir --use-pep517 -e ./src/GroundingDINO && \
+    pip install --no-cache-dir -e ./src/LASER && \
+    pip install --no-cache-dir -e ./vine_hf
+
+RUN cd src/GroundingDINO && \
+    python3 setup.py build_ext --force --inplace
+
+# Copy app
+COPY --chown=user:user . .
+
+EXPOSE 7860
+ENV GRADIO_SERVER_NAME="0.0.0.0"
+CMD ["python3", "app.py"]