Unverified Commit bc9442d1 authored by DaniAndTheWeb's avatar DaniAndTheWeb Committed by GitHub

Merge pull request #2 from DaniAndTheWeb/DaniAndTheWeb-sd-Renoir

Experimental support for AMD Renoir
parents 36364bd7 912285ae
...@@ -105,10 +105,14 @@ fi ...@@ -105,10 +105,14 @@ fi
# Check prerequisites # Check prerequisites
gpu_info=$(lspci 2>/dev/null | grep VGA) gpu_info=$(lspci 2>/dev/null | grep VGA)
if echo "$gpu_info" | grep -qE "Navi (1|2)" case "$gpu_info" in
then *"Navi 1"*|*"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0
export HSA_OVERRIDE_GFX_VERSION=10.3.0 ;;
fi *"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0
;;
*)
;;
esac
if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]] if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]]
then then
export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2" export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2"
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment