FROM pytorch/pytorch:2.12.0-cuda13.2-cudnn9-runtime

RUN pip install --break-system-packages --no-cache-dir \
        "opencv-python>=4.5,<4.12" \
        "numpy>=1.21,<2.3" \
        "gradio" \
        "notebook" \
        "matplotlib"

RUN apt-get update && apt-get install -y \
        ffmpeg \
        libsm6 \
        libxext6 && \
    rm -rf /var/lib/apt/lists/*

WORKDIR /workspace/muggled-sam

CMD ["/bin/bash", "-l"]