Experimenting with PyTorch on Intel architecture
Find a file
2025-09-02 21:21:53 +02:00
.gitignore initial commit 2025-09-02 20:09:03 +02:00
amp-inference.py feat: add inference examples 2025-09-02 21:21:02 +02:00
env.sh feat: add inference examples 2025-09-02 21:21:02 +02:00
fp32-inference.py feat: add inference examples 2025-09-02 21:21:02 +02:00
main.py feat: add inference examples 2025-09-02 21:21:02 +02:00
README.md chore: update readme 2025-09-02 21:21:53 +02:00
torch-compile-inference.py feat: add inference examples 2025-09-02 21:21:02 +02:00

Learn PyTorch

Code experimenting with PyTorch using Intel architecture (i.e., Intel Core Ultra processor with iGPU) or xpu.

After installing ipex-llm which is required to use Intel GPUs (see documentation in Links), you can should have access to conda and be able to import torch normally.

Setup

  1. Run env.sh to activate the conda environment and set
$ . env.sh
  1. (Optional) Confirm if XPU is detected
$ python # go intou the python shell

$ import torch
$ torch.xpu.is_available()
$ torch.xpu.get_device_name()