xpu.txt 538 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
numba == 0.60.0 # v0.61 doesn't support Python 3.9. Required for N-gram speculative decoding
13
nixl==0.3.0 # for PD disaggregation
14
15
--extra-index-url=https://download.pytorch.org/whl/xpu
torch==2.8.0+xpu
16
17
18
torchaudio
torchvision
pytorch-triton-xpu
19
--extra-index-url=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
20
intel-extension-for-pytorch==2.8.10+xpu