diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-09-09 06:10:07 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-09 06:10:07 +0000 |
| commit | 8afabae67d7a5532b33dd961d7e40ac78e015bcd (patch) | |
| tree | 7e86e87ca0fec184a0fd0bd63ea8a64b56e67263 /webui.sh | |
| parent | fccde0c1f72bf3df316f143b37d041b1c4cc5075 (diff) | |
| parent | 737a013377dd698e620f39e405594f7688656af0 (diff) | |
| download | stable-diffusion-webui-gfx803-8afabae67d7a5532b33dd961d7e40ac78e015bcd.tar.gz stable-diffusion-webui-gfx803-8afabae67d7a5532b33dd961d7e40ac78e015bcd.tar.bz2 stable-diffusion-webui-gfx803-8afabae67d7a5532b33dd961d7e40ac78e015bcd.zip | |
Merge pull request #12929 from Beinsezii/dev
WEBUI.SH - Use torch 2.1.0 release candidate for Navi 3
Diffstat (limited to 'webui.sh')
| -rwxr-xr-x | webui.sh | 5 |
1 files changed, 2 insertions, 3 deletions
@@ -141,9 +141,8 @@ case "$gpu_info" in *"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0 ;; *"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \ - export TORCH_COMMAND="pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.6" - # Navi 3 needs at least 5.5 which is only on the nightly chain, previous versions are no longer online (torch==2.1.0.dev-20230614+rocm5.5 torchvision==0.16.0.dev-20230614+rocm5.5 torchaudio==2.1.0.dev-20230614+rocm5.5) - # so switch to nightly rocm5.6 without explicit versions this time + export TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/test/rocm5.6" + # Navi 3 needs at least 5.5 which is only on the torch 2.1.0 release candidates right now ;; *"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0 printf "\n%s\n" "${delimiter}" |