From a8a7bed2b672a66c8e4b7000262efd8ae7cd2806 Mon Sep 17 00:00:00 2001 From: Hongtao Zhang Date: Thu, 14 Nov 2024 16:57:06 -0800 Subject: [PATCH] Dependency - Bump onnxruntime-gpu version from 1.10.0 to 1.12.0 (#663) **Description** Bump onnxruntime-gpu from 1.10.0 to 1.12.0. --------- Co-authored-by: hongtaozhang --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 93d53639c..738095889 100644 --- a/setup.py +++ b/setup.py @@ -215,7 +215,7 @@ def run(self): ], 'ort': [ 'onnx>=1.10.2', - 'onnxruntime-gpu==1.10.0; python_version<"3.10" and platform_machine == "x86_64"', + 'onnxruntime-gpu==1.12.0; python_version<"3.10" and platform_machine == "x86_64"', 'onnxruntime-gpu; python_version>="3.10" and platform_machine == "x86_64"', ], 'nvidia': ['py3nvml>=0.2.6'],