From 10e94c84f6ce4e5f1dff8a2fefc242e9ee687d0c Mon Sep 17 00:00:00 2001 From: Fadi Arafeh <115173828+fadara01@users.noreply.github.com> Date: Fri, 23 Jan 2026 13:13:06 +0000 Subject: [PATCH] [CPU][Feat] Update PyTorch to v2.10 for CPU Backend (#32869) Signed-off-by: Fadi Arafeh Co-authored-by: Li, Jiang --- docker/Dockerfile.cpu | 2 +- requirements/cpu-build.txt | 5 ++--- requirements/cpu.txt | 4 ++-- 3 files changed, 5 insertions(+), 6 deletions(-) diff --git a/docker/Dockerfile.cpu b/docker/Dockerfile.cpu index 2caf1ad14..9bf52d21d 100644 --- a/docker/Dockerfile.cpu +++ b/docker/Dockerfile.cpu @@ -132,7 +132,7 @@ RUN --mount=type=bind,src=requirements/test.in,target=requirements/test.in \ esac; \ }; \ remove_packages_not_supported_on_aarch64 && \ - sed -i 's/^torch==.*/torch==2.9.1/g' requirements/cpu-test.in && \ + sed -i 's/^torch==.*/torch==2.10.0/g' requirements/cpu-test.in && \ sed -i 's/torchaudio.*/torchaudio/g' requirements/cpu-test.in && \ sed -i 's/torchvision.*/torchvision/g' requirements/cpu-test.in && \ uv pip compile requirements/cpu-test.in -o requirements/cpu-test.txt --index-strategy unsafe-best-match --torch-backend cpu diff --git a/requirements/cpu-build.txt b/requirements/cpu-build.txt index a7bd3b17b..3893b0026 100644 --- a/requirements/cpu-build.txt +++ b/requirements/cpu-build.txt @@ -3,9 +3,8 @@ ninja packaging>=24.2 setuptools==77.0.3 # this version can reuse CMake build dir setuptools-scm>=8 -torch==2.9.1+cpu; platform_machine == "x86_64" or platform_machine == "s390x" -torch==2.9.1; platform_system == "Darwin" or platform_machine == "ppc64le" or platform_machine == "aarch64" -scons; platform_machine == "aarch64" # needed to build Arm Compute Library (ACL) +torch==2.10.0+cpu; platform_machine == "x86_64" or platform_machine == "s390x" +torch==2.10.0; platform_machine == "aarch64" or platform_system == "Darwin" or platform_machine == "ppc64le" wheel jinja2>=3.1.6 regex diff --git a/requirements/cpu.txt b/requirements/cpu.txt index 111b8a551..7b3070b42 100644 --- a/requirements/cpu.txt +++ b/requirements/cpu.txt @@ -6,8 +6,8 @@ setuptools==77.0.3 # this version can reuse CMake build dir numba == 0.61.2; platform_machine != "s390x" # Required for N-gram speculative decoding # Dependencies for CPUs -torch==2.9.1+cpu; platform_machine == "x86_64" or platform_machine == "s390x" -torch==2.9.1; platform_system == "Darwin" or platform_machine == "ppc64le" or platform_machine == "aarch64" +torch==2.10.0+cpu; platform_machine == "x86_64" or platform_machine == "s390x" +torch==2.10.0; platform_machine == "aarch64" or platform_system == "Darwin" or platform_machine == "ppc64le" # required for the image processor of minicpm-o-2_6, this must be updated alongside torch torchaudio; platform_machine != "s390x"