Skip to content

Commit 04a1e34

Browse files
dvrogozhNicolasHug
andauthored
doc: add a note on XPU plugin availability (meta-pytorch#1483)
Co-authored-by: Nicolas Hug <contact@nicolas-hug.com> Signed-off-by: Dmitry Rogozhkin <dmitry.v.rogozhkin@intel.com>
1 parent 264a53d commit 04a1e34

1 file changed

Lines changed: 19 additions & 3 deletions

File tree

README.md

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -117,8 +117,6 @@ with encoder.open_file("output.mp4"):
117117
and Windows this will install CPU-only wheels. CPU wheels are available for
118118
Linux (x86_64 and aarch64), macOS, and Windows.
119119

120-
For other versions of PyTorch, refer to the compatibility table below.
121-
122120
### CUDA support
123121

124122
CUDA-enabled wheels are installed by default on Linux. For Windows, you'll need
@@ -164,7 +162,25 @@ default):
164162
pip install torchcodec --index-url=https://download.pytorch.org/whl/cpu
165163
```
166164

167-
### Compatibility
165+
### XPU support
166+
167+
Intel GPUs (XPU) support requires a stand-alone plugin for TorchCodec:
168+
169+
```bash
170+
pip install torchcodec-xpu --extra-index-url=https://download.pytorch.org/whl/xpu
171+
```
172+
173+
Upon installation, the plugin needs to be imported:
174+
175+
```python
176+
import torchcodec_xpu
177+
```
178+
179+
For any XPU-related support, please refer to
180+
https://github.com/intel/torchlib-xpu.
181+
182+
183+
### Compatibility with `torch` versions
168184

169185
The following table indicates the compatibility between versions of
170186
`torchcodec`, `torch` and Python.

0 commit comments

Comments
 (0)