Merge pull request #2 from DaniAndTheWeb/DaniAndTheWeb-sd-Renoir
Experimental support for AMD Renoir
This commit is contained in:
commit
bc9442d1ec
12
webui.sh
12
webui.sh
|
@ -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"
|
||||||
|
|
Loading…
Reference in New Issue
Block a user