Unverified Commit 348abeb9 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #10722 from maybe-hello-world/master 

Download ROCm for AMD GPU only if NVIDIA is not presented
parents ba812b44 bdc37198

webui.sh

+5 −2
Original line number Diff line number Diff line
@@ -124,10 +124,13 @@ case "$gpu_info" in 
    *)

    ;;

esac

if ! echo "$gpu_info" | grep -q "NVIDIA";

then

    if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]]

    then

        export TORCH_COMMAND="pip install torch==2.0.1+rocm5.4.2 torchvision==0.15.2+rocm5.4.2 --index-url https://download.pytorch.org/whl/rocm5.4.2"

    fi

fi



for preq in "${GIT}" "${python_cmd}"

do

CRA Git | Maintained and supported by SUSTech CRA and CCSE