OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

update get start xpu (#143183)

Browse Source 
- Support new Intel client GPU on Windows [Intel® Arc™ B-Series graphics](https://www.intel.com/content/www/us/en/products/docs/discrete-gpus/arc/desktop/b-series/overview.html) and [Intel® Core™ Ultra Series 2 with Intel® Arc™ Graphics](https://www.intel.com/content/www/us/en/products/details/processors/core-ultra.html)
- Support vision/audio prebuilt wheels on Windows
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143183
Approved by: https://github.com/EikanWang, https://github.com/leslie-fang-intel, https://github.com/atalman, https://github.com/malfet

Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com>
This commit is contained in:
ZhaoqiongZ 2025-01-17 06:31:40 +00:00 committed by PyTorch MergeBot
parent fd8e0e3e10
commit 465a1cfe2e
1 changed files with 16 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
docs/source/notes/get_start_xpu.rst
View File

@ -8,23 +8,24 @@ Hardware Prerequisite
:widths: 50 50 :widths: 50 50
:header-rows: 1 :header-rows: 1
* - Validated Hardware * - Supported OS
- Supported OS - Validated Hardware
* - Intel® Data Center GPU Max Series * - Linux
- Linux - Intel® Client GPUs / Intel® Data Center GPU Max Series
* - Intel Client GPU * - Windows
- Windows/Linux - Intel® Client GPUs
* - WSL2 (experimental feature)
- Intel® Client GPUs
Intel GPUs support (Prototype) is ready in PyTorch* 2.5 for Intel® Data Center GPU Max Series and Intel® Client GPUs on both Linux and Windows, which brings Intel GPUs and the SYCL* software stack into the official PyTorch stack with consistent user experience to embrace more AI application scenarios. Intel GPUs support (Prototype) is ready in PyTorch* 2.6 for Intel® Client GPUs and Intel® Data Center GPU Max Series on both Linux and Windows, which brings Intel GPUs and the SYCL* software stack into the official PyTorch stack with consistent user experience to embrace more AI application scenarios.
Software Prerequisite Software Prerequisite
--------------------- ---------------------
Visit `PyTorch Installation Prerequisites for Intel GPUs <https://www.intel.com/content/www/us/en/developer/articles/tool/pytorch-prerequisites-for-intel-gpus.html>`_ for more detailed information regarding: To use PyTorch on Intel GPUs, you need to install the Intel GPUs driver first. For installation guide, visit `Intel GPUs Driver Installation <https://www.intel.com/content/www/us/en/developer/articles/tool/pytorch-prerequisites-for-intel-gpu/2-6.html#driver-installation>`_.
Intel GPU Drivers are sufficient for binary installation, while building from source requires both Intel GPU Drivers and Intel® Deep Learning Essentials. Please refer to `PyTorch Installation Prerequisites for Intel GPUs <https://www.intel.com/content/www/us/en/developer/articles/tool/pytorch-prerequisites-for-intel-gpu/2-6.html>`_ for more information.
#. Intel GPU driver installation
#. Intel support package installation
#. Environment setup
Installation Installation
------------ ------------
@ -32,11 +33,7 @@ Installation
Binaries Binaries
^^^^^^^^ ^^^^^^^^
Platform Linux Now that we have `Intel GPU Driver <https://www.intel.com/content/www/us/en/developer/articles/tool/pytorch-prerequisites-for-intel-gpu/2-6.html#driver-installation>`_ installed, use the following commands to install ``pytorch``, ``torchvision``, ``torchaudio`` on Linux.
""""""""""""""
Now we have all the required packages installed and environment activated. Use the following commands to install ``pytorch``, ``torchvision``, ``torchaudio`` on Linux.
For preview wheels For preview wheels
@ -50,26 +47,13 @@ For nightly wheels
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
Platform Windows
""""""""""""""""
Now we have all the required packages installed and environment activated. Use the following commands to install ``pytorch`` on Windows, build from source for ``torchvision`` and ``torchaudio``.
For preview wheels
.. code-block::
pip3 install torch --index-url https://download.pytorch.org/whl/test/xpu
For nightly wheels
.. code-block::
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
From Source From Source
^^^^^^^^^^^ ^^^^^^^^^^^
Now that we have `Intel GPU Driver and Intel® Deep Learning Essentials <https://www.intel.com/content/www/us/en/developer/articles/tool/pytorch-prerequisites-for-intel-gpu/2-6.html>`_ installed. Follow guides to build ``pytorch``, ``torchvision``, ``torchaudio`` from source.
Build from source for ``torch`` refer to `PyTorch Installation Build from source <https://github.com/pytorch/pytorch?tab=readme-ov-file#from-source>`_. Build from source for ``torch`` refer to `PyTorch Installation Build from source <https://github.com/pytorch/pytorch?tab=readme-ov-file#from-source>`_.
Build from source for ``torchvision`` refer to `Torchvision Installation Build from source <https://github.com/pytorch/vision/blob/main/CONTRIBUTING.md#development-installation>`_. Build from source for ``torchvision`` refer to `Torchvision Installation Build from source <https://github.com/pytorch/vision/blob/main/CONTRIBUTING.md#development-installation>`_.
@ -86,11 +70,7 @@ To check if your Intel GPU is available, you would typically use the following c
import torch import torch
torch.xpu.is_available() # torch.xpu is the API for Intel GPU support torch.xpu.is_available() # torch.xpu is the API for Intel GPU support
If the output is ``False``, double check following steps below. If the output is ``False``, double check driver installation for Intel GPUs.
#. Intel GPU driver installation
#. Intel support package installation
#. Environment setup
Minimum Code Change Minimum Code Change
------------------- -------------------