cpu.txt 738 Bytes
Newer Older
1
# Common dependencies
2
-r common.txt
3

4
# Dependencies for CPUs
5
torch==2.6.0+cpu; platform_machine == "x86_64"
6
7
torch==2.6.0; platform_system == "Darwin"
torch==2.5.1; platform_machine == "ppc64le" or platform_machine == "aarch64"
8
torch==2.7.0.dev20250304; platform_machine == "s390x"
9
10

# required for the image processor of minicpm-o-2_6, this must be updated alongside torch
11
torchaudio; platform_machine != "ppc64le" and platform_machine != "s390x"
12
13
14
torchaudio==2.5.1; platform_machine == "ppc64le"

# required for the image processor of phi3v, this must be updated alongside torch
15
torchvision; platform_machine != "ppc64le"  and platform_machine != "s390x"
16
torchvision==0.20.1; platform_machine == "ppc64le"
17
datasets # for benchmark scripts