diff --git a/README.md b/README.md index 370f64100a6..d9fccb500c8 100644 --- a/README.md +++ b/README.md @@ -135,17 +135,17 @@ 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.0``` +```pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.1``` -This is the command to install the nightly with ROCm 6.0 which might have some performance improvements: +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.1``` +```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.2``` ### NVIDIA Nvidia users should install stable pytorch using this command: -```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121``` +```pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu124``` This is the command to install pytorch nightly instead which might have performance improvements: