File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -135,17 +135,17 @@ Put your VAE in: models/vae
135135### AMD GPUs (Linux only)
136136AMD 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
146146Nvidia 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
150150This is the command to install pytorch nightly instead which might have performance improvements:
151151
You can’t perform that action at this time.
0 commit comments