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.
This commit is contained in:
DaniAndTheWeb 2023-01-19 23:42:12 +01:00 committed by GitHub
parent 36364bd76c
commit 912285ae64
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 4 deletions

View File

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