StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Forcing PyTorch version for AMD GPUs automatic install

Browse Source 
The old code tries to install the newest versions of pytorch, wich is currently 2.0. Forcing it to 1.13.1
This commit is contained in:
DGdev91 2023-04-06 01:36:27 +02:00
parent 22bcc7be42
commit 3a5b47e26e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
webui.sh
View File

@ -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}"