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 0d7690bfaf..136e32677c 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.9.0.dev,<2.10.0 +--extra-index-url https://download.pytorch.org/whl/test/cu130 +torch>=2.9.0,<2.10.0 --extra-index-url https://pypi.ngc.nvidia.com pyyaml dllist diff --git a/pyproject.toml b/pyproject.toml index 63ed81f85f..21978d419d 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.9.0.dev,<2.10.0; platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)", + "torch>=2.9.0,<2.10.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", ] @@ -100,23 +100,23 @@ index-strategy = "unsafe-best-match" [tool.uv.sources] torch = [ - { index = "pytorch-nightly-cu130", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" }, + { index = "pytorch-test-cu130", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" }, { index = "jetson-containers", marker = "platform_machine == 'aarch64' and 'tegra' in platform_release" }, ] torchvision = [ - { index = "pytorch-nightly-cu130", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" }, + { index = "pytorch-test-cu130", marker = "platform_machine != 'aarch64' or (platform_machine == 'aarch64' and 'tegra' not in platform_release)" }, { index = "jetson-containers", marker = "platform_machine == 'aarch64' and 'tegra' in platform_release" }, ] [[tool.uv.index]] -name = "pytorch-nightly-cu130" -url = "https://download.pytorch.org/whl/nightly/cu130" +name = "pytorch-test-cu130" +url = "https://download.pytorch.org/whl/test/cu130" explicit = false -[[tool.uv.index]] -name = "pytorch-nightly-cu129" -url = "https://download.pytorch.org/whl/nightly/cu129" -explicit = false +#[[tool.uv.index]] +#name = "pytorch-nightly-cu129" +#url = "https://download.pytorch.org/whl/nightly/cu129" +#explicit = false [[tool.uv.index]] name = "jetson-containers" diff --git a/setup.py b/setup.py index 291cfe9b97..4a1d9d066b 100644 --- a/setup.py +++ b/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.9.0,<2.10.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.9.0,<2.10.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", diff --git a/tests/py/requirements.txt b/tests/py/requirements.txt index 6c6cbf27dc..92e58c60b1 100644 --- a/tests/py/requirements.txt +++ b/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.24.0.dev,<0.25.0 +--extra-index-url https://download.pytorch.org/whl/test/cu130 +torchvision>=0.24.0,<0.25.0 timm>=1.0.3 \ No newline at end of file