docker

Dockerfile

@@ -1,37 +1,42 @@
-# ---------- Detectron2 v0.6 wheel path (Torch 1.10 + cu113) ----------
-FROM pytorch/pytorch:1.10.2-cuda11.3-cudnn8-runtime
+# ---------- GPU build: CUDA 11.3 + Torch 1.10.2 + D2 0.6 ----------
+FROM nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04
 
-ENV DEBIAN_FRONTEND=noninteractive PIP_NO_CACHE_DIR=1 PYTHONUNBUFFERED=1
+ENV DEBIAN_FRONTEND=noninteractive \
+    PIP_NO_CACHE_DIR=1 \
+    PYTHONUNBUFFERED=1
 
-# Minimal system deps
+# Python 3.8 on 20.04 + build deps
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3-dev build-essential git libgl1 ffmpeg && \
-    rm -rf /var/lib/apt/lists/*
+    python3 python3-pip python3-dev build-essential git libgl1 ffmpeg \
+ && rm -rf /var/lib/apt/lists/*
+RUN python3 -m pip install --upgrade pip
 
 WORKDIR /app
-RUN python3 -m pip install --upgrade pip
 
-# Detectron2 v0.6 prebuilt (matches torch1.10/cu113; see official table)
+# ----- PyTorch 1.10.2 + cu113 (matches D2 0.6 wheel matrix) -----
+RUN pip install --no-cache-dir -f https://download.pytorch.org/whl/cu113/torch_stable.html \
+    torch==1.10.2+cu113 torchvision==0.11.3+cu113
+
+# ----- Detectron2 0.6 prebuilt for torch1.10 + cu113 -----
 RUN pip install --no-cache-dir -f \
   https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html \
   detectron2==0.6
 
-# Your deps (ensure these do NOT pin torch/torchvision/detectron2)
+# ----- Your deps (ensure they DO NOT pin torch/torchvision/detectron2) -----
 COPY requirements.txt /app/requirements.txt
 RUN sed -i '/^torch/d;/^torchvision/d;/^detectron2/d' /app/requirements.txt && \
     pip install --no-cache-dir -r /app/requirements.txt
 
-# App
+# ----- App code + quick sanity print -----
 COPY . /app
-
-# Sanity
-RUN python - <<'PY'
-import sys, torch, detectron2
+RUN python3 - <<'PY'
+import sys, torch
 print("Python:", sys.version)
-print("Torch:", torch.__version__, "CUDA:", torch.version.cuda, "cuda_is_available:", torch.cuda.is_available())
+print("Torch:", torch.__version__, "CUDA:", torch.version.cuda, "cuda_available:", torch.cuda.is_available())
+import detectron2
 print("Detectron2:", detectron2.__version__)
 PY
 
 ENV GRADIO_SERVER_NAME=0.0.0.0 GRADIO_SERVER_PORT=7860
 EXPOSE 7860
-CMD ["python", "app.py"]
+CMD ["python3", "app.py"]