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", ]