Pin tensor-native tracking packages

Author: alexnorell

docker/dockerfiles/Dockerfile.onnx.jetson.7.2.0

@@ -70,6 +70,8 @@ ARG PYTORCH_VERSION=2.10.0
 ARG TORCHVISION_VERSION=0.25.0
 ARG TRITON_VERSION=3.6.0
 ARG TRITON_WHEEL_SHA256=374f52c11a711fd062b4bfbb201fd9ac0a5febd28a96fb41b4a0f51dde3157f4
+ARG SUPERIORVISION_VERSION=0.30.0.dev2
+ARG TRACKTORS_VERSION=2.6.0.dev1
 ARG ONNXRUNTIME_VERSION=1.24.2
 ARG PYTORCH_MAX_JOBS=12
 ARG ORT_BUILD_PARALLEL=8
@@ -298,6 +300,8 @@ RUN python3 -m pip wheel --no-deps --wheel-dir /opt/wheels \
 
 RUN echo "torch==${PYTORCH_VERSION}" > /tmp/overrides.txt && \
     echo "torchvision==${TORCHVISION_VERSION}" >> /tmp/overrides.txt && \
+    echo "superiorvision==${SUPERIORVISION_VERSION}" >> /tmp/overrides.txt && \
+    echo "tracktors==${TRACKTORS_VERSION}" >> /tmp/overrides.txt && \
     uv pip install --system --break-system-packages --index-strategy unsafe-best-match \
     --find-links /opt/wheels \
     --override /tmp/overrides.txt \
@@ -349,6 +353,8 @@ RUN cd /usr/local/lib/python3.12/dist-packages && \
 
 RUN echo "torch==${PYTORCH_VERSION}" > /tmp/runtime-overrides.txt && \
     echo "torchvision==${TORCHVISION_VERSION}" >> /tmp/runtime-overrides.txt && \
+    echo "superiorvision==${SUPERIORVISION_VERSION}" >> /tmp/runtime-overrides.txt && \
+    echo "tracktors==${TRACKTORS_VERSION}" >> /tmp/runtime-overrides.txt && \
     uv pip install --target /opt/python-runtime \
         --index-strategy unsafe-best-match \
         --find-links /opt/wheels \

docker/scripts/verify_jetson_server_runtime.sh

@@ -77,10 +77,12 @@ import flash_attn
 import importlib.metadata
 import importlib.util
 import onnxruntime
+import supervision
 import tensorrt
 import torch
 import torchvision
 import triton
+import trackers
 from torchvision import io as torchvision_io
 
 arch_flags = set(torch._C._cuda_getArchFlags().split())
@@ -97,6 +99,18 @@ assert torchvision.__version__
 assert triton.__version__
 assert torchvision_io
 assert hasattr(torch.ops.image, "decode_jpegs_cuda")
+assert importlib.metadata.version("superiorvision") == "0.30.0.dev2"
+assert importlib.metadata.version("tracktors") == "2.6.0.dev1"
+assert supervision.__file__
+assert trackers.__file__
+assert cv2.__file__.startswith("/opt/opencv/python/"), cv2.__file__
+
+for replaced_distribution in ("supervision", "trackers"):
+    try:
+        installed_version = importlib.metadata.version(replaced_distribution)
+    except importlib.metadata.PackageNotFoundError:
+        continue
+    raise AssertionError((replaced_distribution, installed_version))
 
 torch_requirements = importlib.metadata.requires("torch") or []
 assert not any(
@@ -108,7 +122,10 @@ pip_cuda_distributions = sorted(
     name
     for distribution in importlib.metadata.distributions()
     if (name := distribution.metadata.get("Name"))
-    and name.lower().startswith("nvidia-")
+    and (
+        name.lower().startswith("nvidia-")
+        or name.lower() in {"cuda-bindings", "cuda-toolkit"}
+    )
 )
 assert not pip_cuda_distributions, pip_cuda_distributions