Skip to content

Commit 7016bdb

Browse files
committed
add the minimum support in torch-tensorrt when testing.
1 parent 42fafb7 commit 7016bdb

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

requirements/pytorch/test.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,3 +17,5 @@ fastapi # for `ServableModuleValidator` # not setting version as re-defined in
1717
uvicorn # for `ServableModuleValidator` # not setting version as re-defined in App
1818

1919
tensorboard >=2.9.1, <2.21.0 # for `TensorBoardLogger`
20+
21+
torch-tensorrt >=1.4.0; platform_system == "Linux"

0 commit comments

Comments
 (0)