Skip to content

Commit 69c87b3

Browse files
committed
add torch-tensorrt support when python is lower then py312.
1 parent 7016bdb commit 69c87b3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

requirements/pytorch/test.txt

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

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

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

0 commit comments

Comments
 (0)