Pins TensorRT to a working version to avoid a segfault in 10.8

NVIDIA · Jan 28, 2025 · 597c1e5 · 597c1e5
1 parent 9423503
commit 597c1e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tripy/pyproject.toml b/tripy/pyproject.toml
@@ -7,7 +7,7 @@ readme = "README.md"
 requires-python = ">= 3.9"
 license = { text = "Apache 2.0" }
 dependencies = [
-  "tensorrt~=10.0",
+  "tensorrt (>=10.0,<=10.7)",
   "mlir-tensorrt-compiler==0.1.38+cuda12.trt105",
   "mlir-tensorrt-runtime==0.1.38+cuda12.trt105",
   "colored==2.2.3",