OpenVINO Hardware Improvements (#20071)

* Use OpenVINO directly to detect if devices are available

* Cleanup

* Update OpenVINO

* Cleanup

* Don't try to use OpenVINO when CPU is set as device

* Catch case where input tensor can't be pre-defined

* Cleanup
This commit is contained in:
Nicolas Mowen
2025-09-15 07:35:49 -06:00
committed by GitHub
parent 2d4a0cc584
commit 03fe054078
5 changed files with 63 additions and 29 deletions

View File

@@ -42,9 +42,8 @@ opencv-python-headless == 4.11.0.*
opencv-contrib-python == 4.11.0.*
scipy == 1.14.*
# OpenVino & ONNX
openvino == 2025.1.*
onnxruntime-openvino == 1.22.* ; platform_machine == 'x86_64'
onnxruntime == 1.22.* ; platform_machine == 'aarch64'
openvino == 2025.3.*
onnxruntime == 1.22.*
# Embeddings
transformers == 4.45.*
# Generative AI

View File

@@ -47,7 +47,7 @@ RUN wget -q https://bootstrap.pypa.io/get-pip.py -O get-pip.py \
RUN python3 -m pip config set global.break-system-packages true
COPY docker/rocm/requirements-wheels-rocm.txt /requirements.txt
RUN pip3 uninstall -y onnxruntime-openvino \
RUN pip3 uninstall -y onnxruntime \
&& pip3 install -r /requirements.txt
#######################################################################

View File

@@ -21,7 +21,7 @@ FROM deps AS frigate-tensorrt
ARG PIP_BREAK_SYSTEM_PACKAGES
RUN --mount=type=bind,from=trt-wheels,source=/trt-wheels,target=/deps/trt-wheels \
pip3 uninstall -y onnxruntime-openvino tensorflow-cpu \
pip3 uninstall -y onnxruntime tensorflow-cpu \
&& pip3 install -U /deps/trt-wheels/*.whl
COPY --from=rootfs / /