FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-runtime

RUN pip install \
        "opencv-python>=4.5,<4.12" \
        "numpy>=1.21,<2.3"

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"]