Update Dockerfile

Dockerfile

@@ -10,27 +10,20 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
         libgl1 \
         libglib2.0-0 \
         libgomp1 \
-        gcc \
-        g++ \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# 1) Torch CPU-only (plus léger, HF Spaces gratuits = pas de GPU)
+# 1) Torch CPU-only
 RUN pip install --no-cache-dir \
     torch==2.3.0+cpu \
     torchvision==0.18.0+cpu \
     --index-url https://download.pytorch.org/whl/cpu
 
-# 2) mmengine d'abord (requis par mmcv)
-RUN pip install --no-cache-dir mmengine==0.10.4
+# 2) mmengine + mmcv-lite (pur Python, pas de compilation C++/CUDA)
+RUN pip install --no-cache-dir mmengine==0.10.4 mmcv-lite==2.1.0
 
-# 3) mmcv wheel précompilé CPU pour torch2.3.0 — évite toute compilation
-RUN pip install --no-cache-dir \
-    mmcv==2.1.0 \
-    -f https://download.openmmlab.com/mmcv/dist/cpu/torch2.3/index.html
-
-# 4) Reste des dépendances
+# 3) Reste des dépendances
 RUN pip install --no-cache-dir \
     fastapi==0.111.0 \
     "uvicorn[standard]==0.29.0" \
@@ -42,17 +35,30 @@ RUN pip install --no-cache-dir \
     einops==0.8.0 \
     huggingface_hub==0.23.0
 
-COPY app.py .
+# 4) Cloner Metric3D et patcher l'import mmcv.utils → mmengine
+#    (mono/utils/comm.py importe encore l'ancien mmcv.utils)
+RUN git clone https://github.com/YvanYin/Metric3D.git /metric3d && \
+    # Remplacer "from mmcv.utils import collect_env as collect_base_env"
+    # par un import neutre qui ne plante pas
+    sed -i 's/from mmcv.utils import collect_env as collect_base_env/from mmengine.utils.dl_utils import collect_env as collect_base_env/g' \
+        /metric3d/mono/utils/comm.py && \
+    # Forcer torch.hub à utiliser le repo local déjà cloné
+    mkdir -p /root/.cache/torch/hub && \
+    cp -r /metric3d /root/.cache/torch/hub/yvanyin_metric3d_main
 
-# 5) Vérification des imports + pré-téléchargement des poids
+# 5) Pré-télécharger les poids (le repo est déjà en cache local)
 RUN python -c "\
+import sys; sys.path.insert(0, '/metric3d'); \
 import mmcv; import mmengine; import torch; \
-print('mmcv:', mmcv.__version__, '| mmengine:', mmengine.__version__, '| torch:', torch.__version__); \
-print('Téléchargement metric3d_vit_large ...'); \
-model = torch.hub.load('yvanyin/metric3d', 'metric3d_vit_large', pretrain=True, trust_repo=True); \
+print('mmcv:', mmcv.__version__, '| mmengine:', mmengine.__version__); \
+torch.hub.DEFAULT_CACHE_DIR = '/root/.cache/torch/hub'; \
+model = torch.hub.load('/root/.cache/torch/hub/yvanyin_metric3d_main', \
+    'metric3d_vit_large', pretrain=True, source='local', trust_repo=True); \
 print('Poids en cache OK.'); \
 del model"
 
+COPY app.py .
+
 EXPOSE 7860
 
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]