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'],