Update README.md

This commit is contained in:
Qiacheng Li 2024-12-15 15:33:56 -08:00 committed by GitHub
parent 1731ce0ea5
commit 4109d3f831
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -143,11 +143,11 @@ Put your VAE in: models/vae
### AMD GPUs (Linux only)
AMD users can install rocm and pytorch with pip if you don't have it already installed, this is the command to install the stable version:
```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2```
```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2```
This is the command to install the nightly with ROCm 6.2 which might have some performance improvements:
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2```
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2```
### Intel GPUs (Windows and Linux)