Skip to content

Commit 119df4c

Browse files
committed
Added experimental AMD Strix Halo ROCm support
1 parent 82a973c commit 119df4c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

webui.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -155,6 +155,9 @@ case "$gpu_info" in
155155
*"Navi 3"*) [[ -z "${TORCH_COMMAND}" ]] && \
156156
export TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm5.7"
157157
;;
158+
*"Strix Halo"*) export HSA_OVERRIDE_GFX_VERSION=11.5.1
159+
export TORCH_COMMAND="pip install torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm7.1"
160+
;;
158161
*"Renoir"*) export HSA_OVERRIDE_GFX_VERSION=9.0.0
159162
printf "\n%s\n" "${delimiter}"
160163
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"

0 commit comments

Comments
 (0)