Skip to content

Commit baa6b4d

Browse files
Update manual install instructions.
1 parent d4aeefc commit baa6b4d

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -135,17 +135,17 @@ Put your VAE in: models/vae
135135
### AMD GPUs (Linux only)
136136
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:
137137

138-
```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0```
138+
```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.1```
139139

140-
This is the command to install the nightly with ROCm 6.0 which might have some performance improvements:
140+
This is the command to install the nightly with ROCm 6.2 which might have some performance improvements:
141141

142-
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1```
142+
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2```
143143

144144
### NVIDIA
145145

146146
Nvidia users should install stable pytorch using this command:
147147

148-
```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121```
148+
```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu124```
149149

150150
This is the command to install pytorch nightly instead which might have performance improvements:
151151

0 commit comments

Comments
 (0)