Skip to content

Commit 0552de7

Browse files
Bump pytorch cuda and rocm versions in readme instructions. (Comfy-Org#9273)
1 parent 5828607 commit 0552de7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,7 @@ Put your VAE in: models/vae
203203
### AMD GPUs (Linux only)
204204
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:
205205

206-
```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.3```
206+
```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.4```
207207

208208
This is the command to install the nightly with ROCm 6.4 which might have some performance improvements:
209209

@@ -237,7 +237,7 @@ Additional discussion and help can be found [here](https://github.com/comfyanony
237237

238238
Nvidia users should install stable pytorch using this command:
239239

240-
```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu128```
240+
```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu129```
241241

242242
This is the command to install pytorch nightly instead which might have performance improvements.
243243

0 commit comments

Comments
 (0)