Added intel XPU support

This commit is contained in:
98440
2025-01-20 00:47:57 +08:00
parent 9e51878d18
commit 964064094a
7 changed files with 69 additions and 6 deletions

View File

@@ -32,6 +32,10 @@ pip install torch==2.3.0+cu118 torchaudio==2.3.0+cu118 --extra-index-url https:/
# AMD GPU: install pytorch with your ROCm version, e.g.
pip install torch==2.5.1+rocm6.2 torchaudio==2.5.1+rocm6.2 --extra-index-url https://download.pytorch.org/whl/rocm6.2
# intel GPU: install pytorch with your XPU version, e.g.
# Intel® Deep Learning Essentials or Intel® oneAPI Base Toolkit must be installed
pip install --pre torch torchaudio --index-url https://download.pytorch.org/whl/nightly/xpu
```
Then you can choose from a few options below: