diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-01-15 10:48:25 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-01-15 10:48:25 +0000 |
| commit | ebfdd7baeb660ec66f78e1a0e5e45442025f262d (patch) | |
| tree | 965f2576530f26b1e502e5e69683009586044e7d | |
| parent | d97f467c0d27695d23edad5e4f8898a57e0ccb00 (diff) | |
| parent | ba077e2110cab891a46d14665fb161ce0669f31e (diff) | |
| download | stable-diffusion-webui-gfx803-ebfdd7baeb660ec66f78e1a0e5e45442025f262d.tar.gz stable-diffusion-webui-gfx803-ebfdd7baeb660ec66f78e1a0e5e45442025f262d.tar.bz2 stable-diffusion-webui-gfx803-ebfdd7baeb660ec66f78e1a0e5e45442025f262d.zip | |
Merge pull request #6709 from DaniAndTheWeb/master
Automatic setup for AMD GPUs on Linux
| -rwxr-xr-x | webui.sh | 12 |
1 files changed, 11 insertions, 1 deletions
@@ -165,5 +165,15 @@ else printf "\n%s\n" "${delimiter}" printf "Launching launch.py..." printf "\n%s\n" "${delimiter}" - exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@" + gpu_info=$(lspci | grep VGA) + if echo "$gpu_info" | grep -q "AMD" + then + if [[ -z "${TORCH_COMMAND}" ]] + then + export TORCH_COMMAND="pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2" + fi + HSA_OVERRIDE_GFX_VERSION=10.3.0 exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@" + else + exec "${python_cmd}" "${LAUNCH_SCRIPT}" "$@" + fi fi |