Add tensorrt to dependencies

HolyWu · Mar 30, 2023 · b04d1ee · b04d1ee
1 parent 748d16e
commit b04d1ee
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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",