Skip to content

Commit

Permalink
Merge pull request #2 from DaniAndTheWeb/DaniAndTheWeb-sd-Renoir
Browse files Browse the repository at this point in the history
Experimental support for AMD Renoir
  • Loading branch information
daniandtheweb authored Jan 19, 2023
2 parents 36364bd + 912285a commit bc9442d
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions webui.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down

0 comments on commit bc9442d

Please sign in to comment.