Add RTMLib CUDA backends and staged timing

2026-08-15 10:58:02 -07:00
parent dba60667cb
commit 174ba875dd
8 changed files with 243 additions and 61 deletions
@@ -7,7 +7,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 COPY requirements.lock /tmp/requirements.lock
 RUN python -m pip install --no-cache-dir --upgrade pip==24.3.1 \
    && python -m pip install --no-cache-dir -r /tmp/requirements.lock \
-    && python -m pip check
+    && python -m pip install --no-cache-dir --no-deps rtmlib==0.0.15 \
+    && python -c "import cv2, onnxruntime, rtmlib; print(onnxruntime.get_available_providers())"
 COPY app /opt/tail2-app
 COPY tests /opt/tail2-tests
 ENV PYTHONPATH=/opt/tail2-app PYTHONUNBUFFERED=1