Skip to content

Commit 9d92ddf

Browse files
committed
torch-tensorrt deps.
1 parent 1c45011 commit 9d92ddf

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

requirements/pytorch/test.txt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,4 +18,6 @@ uvicorn # for `ServableModuleValidator` # not setting version as re-defined in
1818

1919
tensorboard >=2.9.1, <2.20.0 # for `TensorBoardLogger`
2020

21-
torch-tensorrt >=1.4.0, <2.8.0; platform_system != "Darwin"
21+
torch-tensorrt >=1.4.0, <2.8.0; platform_system == "Linux" and python_version < "3.11" # Initial linux support starts from 1.4.0
22+
torch-tensorrt >=2.3.0, <2.8.0; platform_system != "Darwin" and python_version < "3.11" # Initial windows support starts from 2.3.0
23+
torch-tensorrt >=2.5.0, <2.8.0; platform_system != "Darwin" and python_version >= "3.12"

0 commit comments

Comments
 (0)