16 lines
323 B
Docker
16 lines
323 B
Docker
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"]
|