delete the trailing .6 of trt

pytorch · Apr 27, 2024 · 1dc194a · 1dc194a
1 parent e3090d0
commit 1dc194a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/dev_dep_versions.yml b/dev_dep_versions.yml
@@ -1,4 +1,4 @@
 __version__: "2.4.0.dev0"
 __cuda_version__: "12.1"
 __cudnn_version__: "8.9"
-__tensorrt_version__: "10.0.1.6"
+__tensorrt_version__: "10.0.1"
diff --git a/py/requirements.txt b/py/requirements.txt
@@ -6,4 +6,4 @@ torch>=2.4.0.dev,<2.5.0
 torchvision>=0.19.0.dev,<0.20.0
 --extra-index-url https://pypi.ngc.nvidia.com
 pyyaml
-tensorrt==10.0.1.6
+tensorrt==10.0.1
diff --git a/pyproject.toml b/pyproject.toml
@@ -8,7 +8,7 @@ requires = [
     "cffi>=1.15.1",
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
-    "tensorrt==10.0.1.6",
+    "tensorrt==10.0.1",
     "torch >=2.4.0.dev,<2.5.0",
     "pybind11==2.6.2",
     "numpy",
@@ -42,7 +42,7 @@ requires-python = ">=3.8"
 keywords = ["pytorch", "torch", "tensorrt", "trt", "ai", "artificial intelligence", "ml", "machine learning", "dl", "deep learning", "compiler", "dynamo", "torchscript", "inference"]
 dependencies = [
     "torch >=2.4.0.dev,<2.5.0",
-    "tensorrt",
+    "tensorrt==10.0.1",
     "packaging>=23",
     "numpy",
     "typing-extensions>=4.7.0",