Add note about cuda version while installing torch

This commit is contained in:
Anirudh M 2025-09-21 04:16:26 -07:00 committed by GitHub
parent d1d9eb94b1
commit 79c8b31a1e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -229,6 +229,8 @@ This is the command to install the Pytorch xpu nightly which might have some per
Nvidia users should install stable pytorch using this command:
> **NOTE**: The --index-url depends on your CUDA version.
```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu129```
This is the command to install pytorch nightly instead which might have performance improvements.