diff --git a/webui.sh b/webui.sh index 89dae163a..a0521ec85 100755 --- a/webui.sh +++ b/webui.sh @@ -152,6 +152,10 @@ case "$gpu_info" in ;; *"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0 ;; + *"Navi 3"*"RX 76"*) [[ -z "${TORCH_COMMAND}" ]] && \ + export TORCH_COMMAND="pip install --extra-index-url https://download.pytorch.org/whl/rocm5.7 torch==2.3.1+rocm5.7 torchvision==0.18.1+rocm5.7 torchaudio==2.3.1+rocm5.7" && \ + export HSA_OVERRIDE_GFX_VERSION=11.0.0 + ;; *"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \ export TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.7" ;;