fix: Docker builder with new Torch versions (#2547)

Author: gs-olive

.github/workflows/docker_builder.yml

@@ -16,7 +16,7 @@ concurrency:
 
 jobs:
   build:
-    runs-on: linux.2xlarge
+    runs-on: linux.4xlarge.nvidia.gpu
 
     # Define key environment variables
     # Container name is of the form torch_tensorrt:<branch_name>

py/requirements.txt

@@ -2,8 +2,8 @@ numpy
 packaging
 pybind11==2.6.2
 --extra-index-url https://download.pytorch.org/whl/nightly/cu121
-torch>=2.2.0.dev,<=2.3.0
-torchvision>=0.17.0.dev,<0.18.0
+torch>=2.3.0.dev,<2.4.0
+torchvision>=0.18.0.dev,<0.19.0
 --extra-index-url https://pypi.ngc.nvidia.com
 tensorrt==8.6.1
 pyyaml

pyproject.toml

@@ -9,8 +9,7 @@ requires = [
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
     "tensorrt>=8.6,<8.7",
-    "torch >=2.2.0.dev,<=2.3.0",
-    #"torch==2.1.0.dev20230731",
+    "torch >=2.3.0.dev,<2.4.0",
     "pybind11==2.6.2",
     "numpy",
 ]
@@ -42,8 +41,7 @@ readme = {file = "py/README.md", content-type = "text/markdown"}
 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.2.0.dev,<=2.3.0",
-    #"torch==2.1.0.dev20230731",
+    "torch >=2.3.0.dev,<2.4.0",
     "tensorrt>=8.6,<8.7",
     "packaging>=23",
     "numpy",
@@ -52,7 +50,7 @@ dependencies = [
 dynamic = ["version"]
 
 [project.optional-dependencies]
-torchvision = ["torchvision >=0.17.dev,<0.18.0"]
+torchvision = ["torchvision >=0.18.dev,<0.19.0"]
 
 [project.urls]
 Homepage = "https://pytorch.org/tensorrt"