Skip to content

Commit

Permalink
Update setup.py: update TRT version (#20557)
Browse files Browse the repository at this point in the history
### Description
As a follow-up of #20506


### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
  • Loading branch information
snnn authored May 4, 2024
1 parent d6280e2 commit 38412b6
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 @@ -230,7 +230,7 @@ def run(self):
"libmigraphx_tf.so.2",
]

tensorrt_dependencies = ["libnvinfer.so.8", "libnvinfer_plugin.so.8", "libnvonnxparser.so.8"]
tensorrt_dependencies = ["libnvinfer.so.10", "libnvinfer_plugin.so.10", "libnvonnxparser.so.10"]

cann_dependencies = ["libascendcl.so", "libacl_op_compiler.so", "libfmk_onnx_parser.so"]

Expand Down

0 comments on commit 38412b6

Please sign in to comment.