We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7748e0e commit 4af33fdCopy full SHA for 4af33fd
setup.py
@@ -775,12 +775,11 @@ def get_sbsa_requirements():
775
sbsa_requirements = base_requirements + ["numpy"]
776
if IS_DLFW_CI:
777
return sbsa_requirements
778
+ # TensorRT does not currently build wheels for Tegra, so we need to use the local tensorrt install from the tarball for thor
779
+ # also due to we use sbsa torch_tensorrt wheel for thor, so when we build sbsa wheel, we need to only include tensorrt dependency.
780
return sbsa_requirements + [
781
"torch>=2.9.0,<2.10.0",
782
"tensorrt>=10.13.0,<10.14.0",
- f"{tensorrt_prefix}>=10.13.0,<10.14.0",
- f"{tensorrt_prefix}-bindings>=10.13.0,<10.14.0",
783
- f"{tensorrt_prefix}-libs>=10.13.0,<10.14.0",
784
]
785
786
0 commit comments