diff --git a/webui.sh b/webui.sh index 24575023..16f200a8 100755 --- a/webui.sh +++ b/webui.sh @@ -82,6 +82,11 @@ gpu_info=$(lspci 2>/dev/null | grep VGA) case "$gpu_info" in *"Navi 1"*|*"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0 ;; + *"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \ + export TORCH_COMMAND="pip install --pre torch torchvision torchaudio --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}" printf "Experimental support for Renoir: make sure to have at least 4GB of VRAM and 10GB of RAM or enable cpu mode: --use-cpu all --no-half" @@ -137,4 +142,4 @@ fi printf "\n%s\n" "${delimiter}" printf "Launching launcher.py..." printf "\n%s\n" "${delimiter}" -exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@" \ No newline at end of file +exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@"