Installation

conda install libuv
# install libjpeg and libpng for torchvision
conda install -c anaconda libjpeg-turbo
conda install -c conda-forge libpng

pip install numpy<2
pip install transformers

# for arc discrete gpu
python -m pip install torch==2.3.1+cxx11.abi torchvision==0.18.1+cxx11.abi torchaudio==2.3.1+cxx11.abi intel-extension-for-pytorch==2.3.110+xpu --extra-index-url <https://pytorch-extension.intel.com/release-whl/stable/xpu/cn/>
# for ultra integrated gpu
python -m pip install torch==2.3.1+cxx11.abi torchvision==0.18.1+cxx11.abi torchaudio==2.3.1+cxx11.abi intel-extension-for-pytorch==2.3.110+xpu --extra-index-url <https://pytorch-extension.intel.com/release-whl/stable/mtl/cn/>

Test

python -c "import torch; import intel_extension_for_pytorch as ipex; print(torch.__version__); print(ipex.__version__); [print(f'[{i}]: {torch.xpu.get_device_properties(i)}') for i in range(torch.xpu.device_count())];"

Reference

https://intel.github.io/intel-extension-for-pytorch/index.html#installation?platform=gpu&version=v2.3.110+xpu&os=windows&package=pip

Usage

https://pytorch.org/tutorials/recipes/intel_extension_for_pytorch.html