keep trt related libs versions >=10.6.0,<10.8.0

pytorch · Dec 31, 2024 · 98d5e93 · 98d5e93
1 parent b79d234
commit 98d5e93
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,7 +8,7 @@ requires = [
     "cffi>=1.15.1",
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
-    "tensorrt-cu12==10.7.0",
+    "tensorrt-cu12>=10.6.0,<10.8.0",
     "torch>=2.6.0.dev,<2.7.0",
     "pybind11==2.6.2",
     "numpy",
@@ -55,9 +55,9 @@ keywords = [
 ]
 dependencies = [
     "torch>=2.6.0.dev,<2.7.0",
-    "tensorrt-cu12==10.7.0",
-    "tensorrt-cu12-bindings==10.7.0",
-    "tensorrt-cu12-libs==10.7.0",
+    "tensorrt-cu12>=10.6.0,<10.8.0",
+    "tensorrt-cu12-bindings>=10.6.0,<10.8.0",
+    "tensorrt-cu12-libs>=10.6.0,<10.8.0",
     "packaging>=23",
     "numpy",
     "typing-extensions>=4.7.0",