xpu.txt 502 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
14
--extra-index-url=https://download.pytorch.org/whl/xpu
torch==2.8.0+xpu
15
16
17
torchaudio
torchvision
pytorch-triton-xpu
18
--extra-index-url=https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
19
intel-extension-for-pytorch==2.8.10+xpu