Dockerfile

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

RUN git clone https://github.com/video-fm/video-sam2.git ./video-sam2 && \
    git clone https://github.com/video-fm/GroundingDINO.git ./groundingdino && \
    git clone https://github.com/kevinxuez/LASER.git ./laser && \
    git clone https://github.com/kevinxuez/vine_hf.git ./vine_hf

# Editable installs
RUN pip install --no-cache-dir -e ./video-sam2 && \
    pip install --no-cache-dir --use-pep517 -e ./groundingdino && \
    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 . .

EXPOSE 7860
ENV GRADIO_SERVER_NAME="0.0.0.0"
CMD ["python3", "app.py"]