diff options
| author | DGdev91 <iperpido91@gmail.com> | 2023-04-05 23:36:27 +0000 |
|---|---|---|
| committer | DGdev91 <iperpido91@gmail.com> | 2023-04-05 23:36:27 +0000 |
| commit | 3a5b47e26e8cb1cd640fedfe7cb5263a588d7088 (patch) | |
| tree | 13878ec83e6d72402547cf66af2e7c3075fcb35e | |
| parent | 22bcc7be428c94e9408f589966c2040187245d81 (diff) | |
| download | stable-diffusion-webui-gfx803-3a5b47e26e8cb1cd640fedfe7cb5263a588d7088.tar.gz stable-diffusion-webui-gfx803-3a5b47e26e8cb1cd640fedfe7cb5263a588d7088.tar.bz2 stable-diffusion-webui-gfx803-3a5b47e26e8cb1cd640fedfe7cb5263a588d7088.zip | |
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
| -rwxr-xr-x | webui.sh | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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}" |