upgrade pytorch from 2.9.0.dev to 2.10.0.dev (#3826)

Author: lanluo-nvidia

MODULE.bazel

@@ -1,6 +1,6 @@
 module(
     name = "torch_tensorrt",
-    version = "2.9.0a0",
+    version = "2.10.0a0",
     repo_name = "org_pytorch_tensorrt",
 )
 

README.md

@@ -5,9 +5,9 @@ Torch-TensorRT
 <h4> Easily achieve the best inference performance for any PyTorch model on the NVIDIA platform. </h4>
 
 [![Documentation](https://img.shields.io/badge/docs-master-brightgreen)](https://nvidia.github.io/Torch-TensorRT/)
-[![pytorch](https://img.shields.io/badge/PyTorch-2.9-green)](https://download.pytorch.org/whl/nightly/cu130)
+[![pytorch](https://img.shields.io/badge/PyTorch-2.10-green)](https://download.pytorch.org/whl/nightly/cu130)
 [![cuda](https://img.shields.io/badge/CUDA-13.0-green)](https://developer.nvidia.com/cuda-downloads)
-[![trt](https://img.shields.io/badge/TensorRT-10.12.0-green)](https://github.com/nvidia/tensorrt)
+[![trt](https://img.shields.io/badge/TensorRT-10.13.0-green)](https://github.com/nvidia/tensorrt)
 [![license](https://img.shields.io/badge/license-BSD--3--Clause-blue)](./LICENSE)
 [![Linux x86-64 Nightly Wheels](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-x86_64.yml/badge.svg?branch=nightly)](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-x86_64.yml)
 [![Linux SBSA Nightly Wheels](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-aarch64.yml/badge.svg?branch=nightly)](https://github.com/pytorch/TensorRT/actions/workflows/build-test-linux-aarch64.yml)
@@ -121,8 +121,8 @@ auto results = trt_mod.forward({input_tensor});
 These are the following dependencies used to verify the testcases. Torch-TensorRT can work with other versions, but the tests are not guaranteed to pass.
 
 - Bazel 8.1.1
-- Libtorch 2.9.0.dev (latest nightly)
-- CUDA 12.8 (CUDA 12.6 on Jetson)
+- Libtorch 2.10.0.dev (latest nightly)
+- CUDA 13.0 (CUDA 12.6 on Jetson)
 - TensorRT 10.13.2.6 (TensorRT 10.3 on Jetson)
 
 ## Deprecation Policy

cpp/include/torch_tensorrt/macros.h

@@ -24,7 +24,7 @@
 #define STR(x) XSTR(x)
 
 #define TORCH_TENSORRT_MAJOR_VERSION 2
-#define TORCH_TENSORRT_MINOR_VERSION 9
+#define TORCH_TENSORRT_MINOR_VERSION 10
 #define TORCH_TENSORRT_PATCH_VERSION 0
 #define TORCH_TENSORRT_VERSION      \
   STR(TORCH_TENSORRT_MAJOR_VERSION) \

py/requirements.txt

@@ -2,7 +2,7 @@ numpy
 packaging
 pybind11==2.6.2
 --extra-index-url https://download.pytorch.org/whl/nightly/cu130
-torch>=2.9.0.dev,<2.10.0
+torch>=2.10.0.dev,<2.11.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.9.0.dev,<2.10.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
+    "torch>=2.10.0.dev,<2.11.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)",
     "torch>=2.8.0,<2.9.0; platform_machine == 'aarch64' and 'tegra' in platform_release",
     "pybind11==2.6.2",
 ]

setup.py

@@ -742,7 +742,7 @@ def get_requirements():
         # standard linux and windows requirements
         requirements = base_requirements + ["numpy"]
         if not IS_DLFW_CI:
-            requirements = requirements + ["torch>=2.9.0.dev,<2.10.0"]
+            requirements = requirements + ["torch>=2.10.0.dev,<2.11.0"]
             if USE_TRT_RTX:
                 requirements = requirements + [
                     "tensorrt_rtx>=1.0.0.21",
@@ -769,7 +769,7 @@ def get_sbsa_requirements():
     if IS_DLFW_CI:
         return sbsa_requirements
     return sbsa_requirements + [
-        "torch>=2.9.0.dev,<2.10.0",
+        "torch>=2.10.0.dev,<2.11.0",
         "tensorrt>=10.13.0,<10.14.0",
         "tensorrt-cu12>=10.13.0,<10.14.0",
         "tensorrt-cu12-bindings>=10.13.0,<10.14.0",

tests/py/requirements.txt

@@ -15,5 +15,5 @@ nvidia-modelopt[all]; python_version >'3.9' and python_version <'3.13'
 # 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.24.0.dev,<0.25.0
+torchvision>=0.25.0.dev,<0.26.0
 timm>=1.0.3

version.txt

@@ -1 +1 @@
-2.9.0a0
+2.10.0a0