Merge branch 'microsoft:main' into microbench/cpu_bw_general

microsoft · Nov 15, 2024 · 6d2de60 · 6d2de60
2 parents f86528d + a8a7bed
commit 6d2de60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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'],