Commit 3a5b47e2 authored by DGdev91's avatar DGdev91
Browse files

Forcing PyTorch version for AMD GPUs automatic install 

The old code tries to install the newest versions of pytorch, wich is currently 2.0. Forcing it to 1.13.1
parent 22bcc7be

webui.sh

+1 −1
Original line number Diff line number Diff line
@@ -118,7 +118,7 @@ case "$gpu_info" in 
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"

    export TORCH_COMMAND="pip install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 --extra-index-url https://download.pytorch.org/whl/rocm5.2"

fi  



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

CRA Git | Maintained and supported by SUSTech CRA and CCSE