From 3a69273f0cbb0981d4f9d44c42382288816b15cd Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Tue, 6 May 2025 08:02:34 -0600 Subject: [PATCH] revert onnx runtime update (#18074) * revert onnx runtime update * Fix docs --- docker/main/requirements-wheels.txt | 6 +++--- docker/tensorrt/requirements-amd64.txt | 2 +- docs/docs/configuration/object_detectors.md | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/docker/main/requirements-wheels.txt b/docker/main/requirements-wheels.txt index 6f905be4a..4ab7e03e6 100644 --- a/docker/main/requirements-wheels.txt +++ b/docker/main/requirements-wheels.txt @@ -37,9 +37,9 @@ opencv-python-headless == 4.11.0.* opencv-contrib-python == 4.11.0.* scipy == 1.14.* # OpenVino & ONNX -openvino == 2025.0.* -onnxruntime-openvino == 1.21.0 ; platform_machine == 'x86_64' -onnxruntime == 1.21.0 ; platform_machine == 'aarch64' +openvino == 2024.4.* +onnxruntime-openvino == 1.20.* ; platform_machine == 'x86_64' +onnxruntime == 1.20.* ; platform_machine == 'aarch64' # Embeddings transformers == 4.45.* # Generative AI diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index 795953359..0e003ca3d 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -13,5 +13,5 @@ nvidia-cudnn-cu12 == 9.5.0.*; platform_machine == 'x86_64' nvidia-cufft-cu11==10.*; platform_machine == 'x86_64' nvidia-cufft-cu12==11.*; platform_machine == 'x86_64' onnx==1.16.*; platform_machine == 'x86_64' -onnxruntime-gpu==1.21.0; platform_machine == 'x86_64' +onnxruntime-gpu==1.20.*; platform_machine == 'x86_64' protobuf==3.20.3; platform_machine == 'x86_64' diff --git a/docs/docs/configuration/object_detectors.md b/docs/docs/configuration/object_detectors.md index 60c1e512d..ae497b41d 100644 --- a/docs/docs/configuration/object_detectors.md +++ b/docs/docs/configuration/object_detectors.md @@ -1069,5 +1069,5 @@ wget -O yolov9-t.pt "https://github.com/WongKinYiu/yolov9/releases/download/v0.1 # prepare and run export script sed -i "s/ckpt = torch.load(attempt_download(w), map_location='cpu')/ckpt = torch.load(attempt_download(w), map_location='cpu', weights_only=False)/g" ./models/experimental.py -python3 export.py --weights ./yolov9-t.pt --imgsz 320 --simplify --include onnx +bin/python3 export.py --weights ./yolov9-t.pt --imgsz 320 --simplify --include onnx ```