From a2b91f3741b56f9318b6681bed395eed406bb790 Mon Sep 17 00:00:00 2001 From: Christian Bayle Date: Fri, 10 Jan 2025 12:31:27 +0100 Subject: [PATCH 1/2] Add support for RX 76xx board (tested on RX 7600 XT) --- webui.sh | 4 ++++ 1 file changed, 4 insertions(+) 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" ;; From 57722f86e6a6e76818887472b18b6c9636c512da Mon Sep 17 00:00:00 2001 From: Christian Bayle Date: Fri, 10 Jan 2025 13:50:03 +0100 Subject: [PATCH 2/2] remove not necessay explicit version for torch torchvision and remove torchaudio --- webui.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/webui.sh b/webui.sh index a0521ec85..ab248682c 100755 --- a/webui.sh +++ b/webui.sh @@ -153,7 +153,7 @@ 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 TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm5.7" && \ export HSA_OVERRIDE_GFX_VERSION=11.0.0 ;; *"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \