Skip to content

Commit

Permalink
Dependency - Bump onnxruntime-gpu version from 1.10.0 to 1.12.0 (#663)
Browse files Browse the repository at this point in the history
**Description**

Bump onnxruntime-gpu from 1.10.0 to 1.12.0.

---------

Co-authored-by: hongtaozhang <[email protected]>
  • Loading branch information
polarG and hongtaozhang authored Nov 15, 2024
1 parent 83ee4eb commit a8a7bed
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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'],
Expand Down

0 comments on commit a8a7bed

Please sign in to comment.