Unverified Commit 912285ae authored by DaniAndTheWeb's avatar DaniAndTheWeb Committed by GitHub
Browse files

Experimental support for Renoir

This adds the GFX version 9.0.0 in order to use Renoir GPUs with at least 4 GB of VRAM (it's possible to increase the virtual VRAM from the BIOS settings of some vendors). This will only work if the remaining ram is at least 12 GB to avoid the system to become unresponsive on launch.).
This change also changes the GPU check to a case statement to be able to add more GPUs efficiently.
parent 36364bd7
Loading
Loading
Loading
Loading
+8 −4
Original line number Diff line number Diff line
@@ -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"