diff options
| author | fraz0815 <l.schimmel83@gmail.com> | 2023-08-20 22:59:30 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-20 22:59:30 +0200 |
| commit | db6c7ff08429c47257970e581f7b2fcfd34ff8e4 (patch) | |
| tree | 14efebd530adf486e59e123363946dd9c8ab0d40 /webui.sh | |
| parent | 68f336bd994bed5442ad95bad6b6ad5564a5409a (diff) | |
Update torch for Navi 31 (7900 XT/XTX)
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
Diffstat (limited to 'webui.sh')
| -rwxr-xr-x | webui.sh | 5 |
1 files changed, 3 insertions, 2 deletions
@@ -141,8 +141,9 @@ 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==2.1.0.dev-20230614+rocm5.5 torchvision==0.16.0.dev-20230614+rocm5.5 --index-url https://download.pytorch.org/whl/nightly/rocm5.5" - # Navi 3 needs at least 5.5 which is only on the nightly chain + 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 ;; *"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0 printf "\n%s\n" "${delimiter}" |