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

Bump pytorch for AMD Users

Browse Source 
So apparently it works now? Before you would get "Pytorch cant use the GPU" but not anymore.
This commit is contained in:
Baptiste Rajaut 2023-05-17 11:01:33 +02:00 committed by GitHub
parent 89f9faa633
commit b3397c2492
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
webui.sh
View File

@ -118,8 +118,8 @@ case "$gpu_info" in
esac esac
if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]] if echo "$gpu_info" | grep -q "AMD" && [[ -z "${TORCH_COMMAND}" ]]
then then
# AMD users will still use torch 1.13 because 2.0 does not seem to work. # Apparently now this works
export TORCH_COMMAND="pip install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 --index-url https://download.pytorch.org/whl/rocm5.2" 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}" for preq in "${GIT}" "${python_cmd}"