ayo/pytorch-intel-experiments
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
Experimenting with PyTorch on Intel architecture
6 commits 1 branch 0 tags 37 KiB
Python 96.7%
Shell 3.3%
Find a file
Ayo 3b9fe4de06 feat: add inference examples
2025-09-02 21:21:02 +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 feat: add inference examples 2025-09-02 21:21:02 +02:00
torch-compile-inference.py feat: add inference examples 2025-09-02 21:21:02 +02:00

README.md

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()

Links