diff --git a/requirements/cpu-build.txt b/requirements/cpu-build.txt index 37f072202bd7..41c03c355341 100644 --- a/requirements/cpu-build.txt +++ b/requirements/cpu-build.txt @@ -6,7 +6,7 @@ packaging>=24.2 setuptools>=77.0.3,<80.0.0 setuptools-scm>=8 --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0+cpu +torch==2.8.0 wheel jinja2>=3.1.6 regex diff --git a/requirements/cpu.txt b/requirements/cpu.txt index a48cb9fde000..2dfbaabf6b81 100644 --- a/requirements/cpu.txt +++ b/requirements/cpu.txt @@ -8,7 +8,7 @@ numba == 0.61.2; python_version > '3.9' and platform_machine != "s390x" packaging>=24.2 setuptools>=77.0.3,<80.0.0 --extra-index-url https://download.pytorch.org/whl/cpu -torch==2.6.0+cpu; platform_machine == "x86_64" # torch>2.6.0+cpu has performance regression on x86 platform, see https://github.com/pytorch/pytorch/pull/151218 +torch==2.8.0; platform_machine == "x86_64" # torch>2.6.0+cpu has performance regression on x86 platform, see https://github.com/pytorch/pytorch/pull/151218 torch==2.8.0; platform_system == "Darwin" torch==2.8.0; platform_machine == "ppc64le" or platform_machine == "aarch64" diff --git a/requirements/test.txt b/requirements/test.txt index 39040f210b2f..84f62fa9cc38 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -1078,7 +1078,7 @@ tomli==2.2.1 # via schemathesis tomli-w==1.2.0 # via schemathesis -torch==2.8.0+cu128 +torch==2.8.0 # via # -r requirements/test.in # accelerate