diff --git a/pyproject.toml b/pyproject.toml index 4e0d73b..e4f30cc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -20,6 +20,7 @@ classifiers = [ dependencies = [ "numpy>=1.24.2", "requests>=2.28.2", + "tensorrt>=8.5.3.1", "torch>=1.13.1", "torch-tensorrt-fx-only>=1.3.0", "tqdm>=4.65.0",