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

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
...@@ -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