suport a range of versions for torch and tensorrt (#2897)

pytorch · Jun 7, 2024 · 52ba6f1 · 52ba6f1
1 parent 88ee123
commit 52ba6f1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,8 +8,8 @@ requires = [
     "cffi>=1.15.1",
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
-    "tensorrt",
-    "torch==2.3.1",
+    "tensorrt>=10.0.0,<10.1.0",
+    "torch>=2.3.0,<2.3.2",
     "pybind11==2.6.2",
     "numpy",
 ]
@@ -41,8 +41,8 @@ readme = {file = "py/README.md", content-type = "text/markdown"}
 requires-python = ">=3.8"
 keywords = ["pytorch", "torch", "tensorrt", "trt", "ai", "artificial intelligence", "ml", "machine learning", "dl", "deep learning", "compiler", "dynamo", "torchscript", "inference"]
 dependencies = [
-    "torch==2.3.1",
-    "tensorrt",
+    "torch>=2.3.0,<2.3.2",
+    "tensorrt>=10.0.0,<10.1.0",
     "packaging>=23",
     "numpy",
     "typing-extensions>=4.7.0",