Skip to content

GitLab

Unverified Commit 393de22d authored by Fadi Arafeh's avatar Fadi Arafeh Committed by GitHub
Browse files

[fix] Update torch version in cpu-build.txt for AArch64/ppc64le and Darwin (#25579) 


Signed-off-by: default avatarFadi Arafeh <fadi.arafeh@arm.com>
parent 1260180c
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
requirements/cpu-build.txt
View file @ 393de22d
...@@ -6,7 +6,8 @@ packaging>=24.2 ...@@ -6,7 +6,8 @@ packaging>=24.2
setuptools>=77.0.3,<80.0.0 setuptools>=77.0.3,<80.0.0
setuptools-scm>=8 setuptools-scm>=8
--extra-index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu
torch==2.6.0+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 == "ppc64le" or platform_machine == "aarch64" or platform_system == "Darwin"
wheel wheel
jinja2>=3.1.6 jinja2>=3.1.6
regex regex
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment