diff --git a/requirements/cpu.txt b/requirements/cpu.txt index fc090837..1205c098 100644 --- a/requirements/cpu.txt +++ b/requirements/cpu.txt @@ -15,3 +15,6 @@ torchaudio==2.6.0; platform_machine == "ppc64le" torchvision; platform_machine != "ppc64le" and platform_machine != "s390x" torchvision==0.21.0; platform_machine == "ppc64le" datasets # for benchmark scripts + +# cpu cannot use triton 3.3.0 +triton==3.2.0