diff --git a/README.md b/README.md index b1689af88a..0c9291272f 100644 --- a/README.md +++ b/README.md @@ -169,7 +169,7 @@ the command `npm install -g pnpm` if needed) _For Linux with an AMD GPU:_ ```sh - pip install InvokeAI --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 + pip install InvokeAI --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.6 ``` _For non-GPU systems:_ diff --git a/docs/installation/010_INSTALL_AUTOMATED.md b/docs/installation/010_INSTALL_AUTOMATED.md index f6e799aac0..03b1e9be89 100644 --- a/docs/installation/010_INSTALL_AUTOMATED.md +++ b/docs/installation/010_INSTALL_AUTOMATED.md @@ -477,7 +477,7 @@ Then type the following commands: === "AMD System" ```bash - pip install torch torchvision --force-reinstall --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 + pip install torch torchvision --force-reinstall --extra-index-url https://download.pytorch.org/whl/rocm5.6 ``` ### Corrupted configuration file diff --git a/docs/installation/020_INSTALL_MANUAL.md b/docs/installation/020_INSTALL_MANUAL.md index 1443433362..a68039406c 100644 --- a/docs/installation/020_INSTALL_MANUAL.md +++ b/docs/installation/020_INSTALL_MANUAL.md @@ -154,7 +154,7 @@ manager, please follow these steps: === "ROCm (AMD)" ```bash - pip install InvokeAI --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 + pip install InvokeAI --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.6 ``` === "CPU (Intel Macs & non-GPU systems)" @@ -345,7 +345,7 @@ installation protocol (important!) === "ROCm (AMD)" ```bash - pip install -e . --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 + pip install -e . --use-pep517 --extra-index-url https://download.pytorch.org/whl/rocm5.6 ``` === "CPU (Intel Macs & non-GPU systems)" diff --git a/docs/installation/030_INSTALL_CUDA_AND_ROCM.md b/docs/installation/030_INSTALL_CUDA_AND_ROCM.md index ca33eb5fed..48c1c526ca 100644 --- a/docs/installation/030_INSTALL_CUDA_AND_ROCM.md +++ b/docs/installation/030_INSTALL_CUDA_AND_ROCM.md @@ -134,7 +134,7 @@ recipes are available When installing torch and torchvision manually with `pip`, remember to provide the argument `--extra-index-url -https://download.pytorch.org/whl/rocm5.4.2` as described in the [Manual +https://download.pytorch.org/whl/rocm5.6` as described in the [Manual Installation Guide](020_INSTALL_MANUAL.md). This will be done automatically for you if you use the installer diff --git a/installer/lib/installer.py b/installer/lib/installer.py index 3bca5e108c..5227f181d8 100644 --- a/installer/lib/installer.py +++ b/installer/lib/installer.py @@ -455,7 +455,7 @@ def get_torch_source() -> (Union[str, None], str): optional_modules = "[onnx]" if OS == "Linux": if device == "rocm": - url = "https://download.pytorch.org/whl/rocm5.4.2" + url = "https://download.pytorch.org/whl/rocm5.6" elif device == "cpu": url = "https://download.pytorch.org/whl/cpu"