Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 12 additions & 0 deletions _additional_platforms/xpu.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
{
"name": "XPU",
"support_channel": "https://github.com/pytorch/pytorch",
"stable": {
"linux": "pip3 install torch torchvision --index-url https://download.pytorch.org/whl/xpu",
"windows": "pip3 install torch torchvision --index-url https://download.pytorch.org/whl/xpu"
},
"preview": {
"linux": "pip3 install torch torchvision --pre --index-url https://download.pytorch.org/whl/nightly/xpu",
"windows": "pip3 install torch torchvision --pre --index-url https://download.pytorch.org/whl/nightly/xpu"
}
}
56 changes: 56 additions & 0 deletions _get_started/additional_platforms/xpu.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,56 @@
# Installing on Intel GPU (XPU) Platform

XPU is a PyTorch device backend designed to support hardware acceleration on Intel GPUs. Key technical features:

* Native support for FP32, BF16, FP16, and Automatic Mixed Precision (AMP)
* Extensions of operator set through custom SYCL kernels
* Graph compilation
* Distributed training (through `XCCL`)

## Prerequisites

### Hardware Requirements

* Intel Client GPU:

* Intel® Arc A-Series Graphics (CodeName: Alchemist)
* Intel® Arc B-Series Graphics (CodeName: Battlemage)
* Intel® Core™ Ultra Processors with Intel® Arc™ Graphics (CodeName: Meteor Lake-H)
* Intel® Core™ Ultra Processors (Series 2) with Intel® Arc™ Graphics (CodeName: Arrow Lake-H)
* Intel® Core™ Ultra Mobile Processors (Series 2) with Intel® Arc™ Graphics (CodeName: Lunar Lake)
Comment thread
dvrogozh marked this conversation as resolved.
* Intel® Core™ Ultra Mobile Processors (Series 3) with Intel® Arc™ Graphics (CodeName: Panther Lake)

* Intel Data Center GPU:

* Intel® Data Center GPU Max Series (CodeName: Ponte Vecchio)

### Software Requirements

* [Intel GPU Driver](https://www.intel.com/content/www/us/en/developer/articles/tool/pytorch-prerequisites-for-intel-gpu.html)
* Python 3.10 or later

## Installation

### pip

Use the pip package manager to install PyTorch with XPU support. Select your preferred options in the selector above to get the installation command.

## Verification

To ensure that PyTorch was installed correctly with XPU support, run the following code:

```python
import torch
print(torch.__version__)

# Check XPU availability
if torch.xpu.is_available():
print("XPU is available!")
print(f"XPU devices: {torch.xpu.device_count()}")
else:
print("XPU is not available.")
```

## Documentation

For more information, please visit the [Getting Started on Intel GPU](https://docs.pytorch.org/docs/stable/notes/get_start_xpu.html).