xpu.txt 665 Bytes
Newer Older
1
# Common dependencies
2
-r common.txt
3

4
ray>=2.9
5
cmake>=3.26.1
6
packaging>=24.2
7
setuptools-scm>=8
8
setuptools>=77.0.3,<80.0.0
9
wheel
10
jinja2>=3.1.6
11
datasets # for benchmark scripts
12

13
torch==2.7.0+xpu
14
15
16
17
torchaudio
torchvision
pytorch-triton-xpu
--extra-index-url=https://download.pytorch.org/whl/xpu
18

19
20
# Please refer xpu doc, we need manually install intel-extension-for-pytorch 2.6.10+xpu due to there are some conflict dependencies with torch 2.6.0+xpu
# FIXME: This will be fix in ipex 2.7. just leave this here for awareness.
21
22
intel-extension-for-pytorch==2.7.10+xpu
oneccl_bind_pt==2.7.0+xpu
23
--extra-index-url=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/