Experimenting with PyTorch on Intel architecture
inference-examples | ||
.gitignore | ||
check-xpu.py | ||
env.sh | ||
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 Setup), you will have access to conda
and be able to import torch
normally.
Setup
$ git clone https://git.ayo.run/ayo/learn-pytorch
- Run
env.sh
to activate the conda environment and set
$ . env.sh
- (Optional) Confirm if XPU is detected
$ python # go intou the python shell
$ import torch
$ torch.xpu.is_available()
$ torch.xpu.get_device_name()