Skip to content

Commit 4af33fd

Browse files
fix thor tensorrt dependency issue (#3843)
1 parent 7748e0e commit 4af33fd

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

setup.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -775,12 +775,11 @@ def get_sbsa_requirements():
775775
sbsa_requirements = base_requirements + ["numpy"]
776776
if IS_DLFW_CI:
777777
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.
778780
return sbsa_requirements + [
779781
"torch>=2.9.0,<2.10.0",
780782
"tensorrt>=10.13.0,<10.14.0",
781-
f"{tensorrt_prefix}>=10.13.0,<10.14.0",
782-
f"{tensorrt_prefix}-bindings>=10.13.0,<10.14.0",
783-
f"{tensorrt_prefix}-libs>=10.13.0,<10.14.0",
784783
]
785784

786785

0 commit comments

Comments
 (0)