Update command to install AMD stable linux pytorch. (#12437)
parent
ae79e33345
commit
e03fe8b591
|
|
@ -227,7 +227,7 @@ Put your VAE in: models/vae
|
|||
|
||||
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.4```
|
||||
```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm7.1```
|
||||
|
||||
This is the command to install the nightly with ROCm 7.1 which might have some performance improvements:
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue