From dfc94228fcdc7bebcb58b57b7a6a06fa1111a62c Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Tue, 17 Sep 2024 10:14:39 -0700 Subject: [PATCH] upgrade torch from 2.5.0.dev to 2.6.0.dev in main branch --- py/requirements.txt | 2 +- pyproject.toml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/py/requirements.txt b/py/requirements.txt index 15d359fbc3..361afab365 100644 --- a/py/requirements.txt +++ b/py/requirements.txt @@ -2,7 +2,7 @@ numpy packaging pybind11==2.6.2 --extra-index-url https://download.pytorch.org/whl/nightly/cu124 -torch>=2.5.0.dev,<2.6.0 +torch>=2.6.0.dev,<2.7.0 torchvision>=0.20.0.dev,<0.21.0 --extra-index-url https://pypi.ngc.nvidia.com pyyaml diff --git a/pyproject.toml b/pyproject.toml index bc6c6b60cf..26f8e97b80 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,7 +9,7 @@ requires = [ "typing-extensions>=4.7.0", "future>=0.18.3", "tensorrt-cu12==10.3.0", - "torch >=2.5.0.dev,<2.6.0", + "torch >=2.6.0.dev,<2.7.0", "pybind11==2.6.2", "numpy", ]