Skip to content

Commit 52ba6f1

Browse files
suport a range of versions for torch and tensorrt (#2897)
1 parent 88ee123 commit 52ba6f1

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pyproject.toml

+4-4
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@ requires = [
88
"cffi>=1.15.1",
99
"typing-extensions>=4.7.0",
1010
"future>=0.18.3",
11-
"tensorrt",
12-
"torch==2.3.1",
11+
"tensorrt>=10.0.0,<10.1.0",
12+
"torch>=2.3.0,<2.3.2",
1313
"pybind11==2.6.2",
1414
"numpy",
1515
]
@@ -41,8 +41,8 @@ readme = {file = "py/README.md", content-type = "text/markdown"}
4141
requires-python = ">=3.8"
4242
keywords = ["pytorch", "torch", "tensorrt", "trt", "ai", "artificial intelligence", "ml", "machine learning", "dl", "deep learning", "compiler", "dynamo", "torchscript", "inference"]
4343
dependencies = [
44-
"torch==2.3.1",
45-
"tensorrt",
44+
"torch>=2.3.0,<2.3.2",
45+
"tensorrt>=10.0.0,<10.1.0",
4646
"packaging>=23",
4747
"numpy",
4848
"typing-extensions>=4.7.0",

0 commit comments

Comments
 (0)