2.12 torch-tensorrt release cut (#4206)

Author: lanluo-nvidia

docker/dist-build.sh

@@ -4,7 +4,7 @@ set -x
 
 TOP_DIR=$(cd $(dirname $0); pwd)/..
 
-BUILD_CMD="python -m pip wheel .  --extra-index-url https://download.pytorch.org/whl/nightly/cu130 -w dist"
+BUILD_CMD="python -m pip wheel .  --extra-index-url https://download.pytorch.org/whl/test/cu130 -w dist"
 
 # TensorRT restricts our pip version
 cd ${TOP_DIR} \

py/requirements.txt

@@ -1,8 +1,8 @@
 numpy
 packaging
 pybind11==2.6.2
---extra-index-url https://download.pytorch.org/whl/nightly/cu130
-torch>=2.12.0.dev,<2.13.0
+--extra-index-url https://download.pytorch.org/whl/test/cu130
+torch>=2.12.0,<2.13.0
 --extra-index-url https://pypi.ngc.nvidia.com
 pyyaml
 dllist

pyproject.toml

@@ -6,7 +6,7 @@ requires = [
     "ninja>=1.11.0",
     "pyyaml>=6.0",
     "cffi>=1.15.1",
-    "torch>=2.12.0.dev,<2.13.0",
+    "torch>=2.12.0,<2.13.0",
     "pybind11==2.6.2",
 ]
 build-backend = "setuptools.build_meta"
@@ -152,10 +152,10 @@ conflicts = [
 
 [tool.uv.sources]
 torch = [
-    { index = "pytorch-nightly-cu130" },
+    { index = "pytorch-test-cu130" },
 ]
 torchvision = [
-    { index = "pytorch-nightly-cu130" },
+    { index = "pytorch-test-cu130" },
 ]
 
 [[tool.uv.index]]

setup.py

@@ -780,7 +780,7 @@ def get_sbsa_requirements(base_requirements):
         # TensorRT does not currently build wheels for Tegra, so we need to use the local tensorrt install from the tarball for thor
         # also due to we use sbsa torch_tensorrt wheel for thor, so when we build sbsa wheel, we need to only include tensorrt dependency.
         return requirements + [
-            "torch>=2.12.0.dev,<2.13.0",
+            "torch>=2.12.0,<2.13.0",
             "tensorrt>=10.16.1,<10.17.0",
         ]
 
@@ -791,7 +791,7 @@ def get_x86_64_requirements(base_requirements):
     if IS_DLFW_CI:
         return requirements
     else:
-        requirements = requirements + ["torch>=2.12.0.dev,<2.13.0"]
+        requirements = requirements + ["torch>=2.12.0,<2.13.0"]
         if USE_TRT_RTX:
             return requirements + [
                 "tensorrt_rtx>=1.4.0.76",

tests/py/requirements.txt

@@ -14,6 +14,6 @@ nvidia-modelopt[all]; python_version >'3.9' and python_version <'3.13'
 # flashinfer-python is not supported for python version 3.13 or higher
 # flashinfer-python is broken on python 3.9 at the moment, so skip it for now
 flashinfer-python; python_version >'3.9' and python_version <'3.13'
---extra-index-url https://download.pytorch.org/whl/nightly/cu130
-torchvision>=0.26.0.dev,<0.27.0
+--extra-index-url https://download.pytorch.org/whl/test/cu130
+torchvision>=0.26.0,<0.27.0
 timm>=1.0.3