19 lines
426 B
Docker
19 lines
426 B
Docker
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"]
|