Fix Docker scripts for muggled-sam

2026-06-18 16:53:14 +08:00
parent e17bef0f89
commit e89601d835
7 changed files with 13 additions and 6 deletions
@@ -1,8 +1,11 @@
-FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-runtime
+FROM pytorch/pytorch:2.12.0-cuda13.2-cudnn9-runtime

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

 RUN apt-get update && apt-get install -y \
        ffmpeg \