aboutsummaryrefslogtreecommitdiff
path: root/webui.sh
diff options
context:
space:
mode:
authorDaniAndTheWeb <57776841+DaniAndTheWeb@users.noreply.github.com>2023-01-19 23:55:56 +0100
committerGitHub <noreply@github.com>2023-01-19 23:55:56 +0100
commitbc9442d1ecc6804b7009a1320e6f68a9cf515ef0 (patch)
treec47f6ab5508f9435f60887b8ee4aaed5d54ce27b /webui.sh
parent36364bd76c4634820e08070a287f0a5ad27c35f6 (diff)
parent912285ae64e4e1186feb54caf82b4a0b11c6cb7f (diff)
Merge pull request #2 from DaniAndTheWeb/DaniAndTheWeb-sd-Renoir
Experimental support for AMD Renoir
Diffstat (limited to 'webui.sh')
-rwxr-xr-xwebui.sh12
1 files changed, 8 insertions, 4 deletions
diff --git a/webui.sh b/webui.sh
index 91c95e47..27933c04 100755
--- a/webui.sh
+++ b/webui.sh
@@ -105,10 +105,14 @@ fi
# Check prerequisites
gpu_info=$(lspci 2>/dev/null | grep VGA)
-if echo "$gpu_info" | grep -qE "Navi (1|2)"
-then
- export HSA_OVERRIDE_GFX_VERSION=10.3.0
-fi
+case "$gpu_info" in
+ *"Navi 1"*|*"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0
+ ;;
+ *"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0
+ ;;
+ *)
+ ;;
+esac
if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]]
then
export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2"