Skip to content

Commit 8bcc5a3

Browse files
committed
xpu: add XPU to additional pages
XPU is a PyTorch device backend designed to support hardware acceleration on Intel GPUs. Signed-off-by: Dmitry Rogozhkin <dmitry.v.rogozhkin@intel.com>
1 parent 5598012 commit 8bcc5a3

2 files changed

Lines changed: 68 additions & 0 deletions

File tree

_additional_platforms/xpu.json

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
{
2+
"name": "XPU",
3+
"support_channel": "https://github.com/pytorch/pytorch",
4+
"stable": {
5+
"linux": "pip3 install torch torchvision --index-url https://download.pytorch.org/whl/xpu",
6+
"windows": "pip3 install torch torchvision --index-url https://download.pytorch.org/whl/xpu"
7+
},
8+
"preview": {
9+
"linux": "pip3 install torch torchvision --pre --index-url https://download.pytorch.org/whl/nightly/xpu",
10+
"windows": "pip3 install torch torchvision --pre --index-url https://download.pytorch.org/whl/nightly/xpu"
11+
}
12+
}
Lines changed: 56 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,56 @@
1+
# Installing on Intel GPU (XPU) Platform
2+
3+
XPU is a PyTorch device backend designed to support hardware acceleration on Intel GPUs. Key technical features:
4+
5+
* Native support for FP32, BF16, FP16, and Automatic Mixed Precision (AMP)
6+
* Extensions of operator set through custom SYCL kernels
7+
* Graph compilation
8+
* Distributed training (through `XCCL`)
9+
10+
## Prerequisites
11+
12+
### Hardware Requirements
13+
14+
* Intel Client GPU:
15+
16+
* Intel® Arc A-Series Graphics (CodeName: Alchemist)
17+
* Intel® Arc B-Series Graphics (CodeName: Battlemage)
18+
* Intel® Core™ Ultra Processors with Intel® Arc™ Graphics (CodeName: Meteor Lake-H)
19+
* Intel® Core™ Ultra Processors (Series 2) with Intel® Arc™ Graphics (CodeName: Arrow Lake-H)
20+
* Intel® Core™ Ultra Mobile Processors (Series 2) with Intel® Arc™ Graphics (CodeName: Lunar Lake)
21+
* Intel® Core™ Ultra Mobile Processors (Series 3) with Intel® Arc™ Graphics (CodeName: Panther Lake)
22+
23+
* Intel Data Center GPU:
24+
25+
* Intel® Data Center GPU Max Series (CodeName: Ponte Vecchio)
26+
27+
### Software Requirements
28+
29+
* [Intel GPU Driver](https://www.intel.com/content/www/us/en/developer/articles/tool/pytorch-prerequisites-for-intel-gpu.html)
30+
* Python 3.10 or later
31+
32+
## Installation
33+
34+
### pip
35+
36+
Use the pip package manager to install PyTorch with XPU support. Select your preferred options in the selector above to get the installation command.
37+
38+
## Verification
39+
40+
To ensure that PyTorch was installed correctly with XPU support, run the following code:
41+
42+
```python
43+
import torch
44+
print(torch.__version__)
45+
46+
# Check XPU availability
47+
if torch.xpu.is_available():
48+
print("XPU is available!")
49+
print(f"XPU devices: {torch.xpu.device_count()}")
50+
else:
51+
print("XPU is not available.")
52+
```
53+
54+
## Documentation
55+
56+
For more information, please visit the [Getting Started on Intel GPU](https://docs.pytorch.org/docs/stable/notes/get_start_xpu.html).

0 commit comments

Comments
 (0)