From db6c7ff08429c47257970e581f7b2fcfd34ff8e4 Mon Sep 17 00:00:00 2001 From: fraz0815 Date: Sun, 20 Aug 2023 22:59:30 +0200 Subject: 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 --- webui.sh | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'webui.sh') diff --git a/webui.sh b/webui.sh index cb8b9d14..b4c584e9 100755 --- a/webui.sh +++ b/webui.sh @@ -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}" -- cgit v1.2.1