Added experimental AMD Strix Halo ROCm support

This commit is contained in:
Jason Lethbridge 2025-11-21 06:13:04 +00:00
parent 82a973c043
commit 119df4c775

View file

@ -155,6 +155,9 @@ case "$gpu_info" in
*"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \
export TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.7"
;;
*"Strix Halo"*) export HSA_OVERRIDE_GFX_VERSION=11.5.1
export TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.1"
;;
*"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0
printf "\n%s\n" "${delimiter}"
printf "Experimental support for Renoir: make sure to have at least 4GB of VRAM and 10GB of RAM or enable cpu mode: --use-cpu all --no-half"