From 63786273fdaa160078032ea73ff8bca4560e78b7 Mon Sep 17 00:00:00 2001 From: Lan Luo Date: Thu, 11 Jun 2026 09:51:49 -0700 Subject: [PATCH] 2.13 release cut --- docker/dist-build.sh | 2 +- py/requirements.txt | 4 ++-- pyproject.toml | 8 ++++---- setup.py | 4 ++-- 4 files changed, 9 insertions(+), 9 deletions(-) diff --git a/docker/dist-build.sh b/docker/dist-build.sh index 7ade4f10fd..66e560d306 100755 --- a/docker/dist-build.sh +++ b/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} \ diff --git a/py/requirements.txt b/py/requirements.txt index 52964a32f0..c1654c6143 100644 --- a/py/requirements.txt +++ b/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.13.0.dev,<2.14.0 +--extra-index-url https://download.pytorch.org/whl/test/cu130 +torch>=2.13.0,<2.14.0 --extra-index-url https://pypi.ngc.nvidia.com pyyaml dllist diff --git a/pyproject.toml b/pyproject.toml index 770b0f5c57..c9c262e797 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ requires = [ "ninja>=1.11.0", "pyyaml>=6.0", "cffi>=1.15.1", - "torch>=2.13.0.dev,<2.14.0", + "torch>=2.13.0,<2.14.0", "pybind11==2.6.2", ] build-backend = "setuptools.build_meta" @@ -90,7 +90,7 @@ test = [ test-ext = [ "timm>=1.0.3", "transformers>=5.0.0", - "torchvision>=0.27.0.dev,<0.28.0", + "torchvision>=0.27.0,<0.28.0", "flashinfer-python; python_version >'3.9' and python_version <'3.13'", ] @@ -161,10 +161,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]] diff --git a/setup.py b/setup.py index 41e9370cb9..619391d72f 100644 --- a/setup.py +++ b/setup.py @@ -837,7 +837,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.13.0.dev,<2.14.0", + "torch>=2.13.0,<2.14.0", "tensorrt>=11.0.0,<11.1.0", ] @@ -849,7 +849,7 @@ def get_x86_64_requirements(base_requirements): return requirements else: requirements = requirements + [ - "torch>=2.13.0.dev,<2.14.0", + "torch>=2.13.0,<2.14.0", ] if USE_TRT_RTX: return requirements + [