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:
parent
36364bd76c
commit
912285ae64
12
webui.sh
12
webui.sh
|
@ -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"
|
||||
|
|
Loading…
Reference in New Issue