DiffusionAi/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui synced 2025-01-09 09:57:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Skip force pyton and pytorch ver if TORCH_COMMAND already set

Browse Source
This commit is contained in:
DGdev91 2023-06-06 15:43:32 +02:00
parent 8646768801
commit 62860c221e
1 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
webui.sh
View File

@ -114,16 +114,19 @@ fi
# Check prerequisites # Check prerequisites
gpu_info=$(lspci 2>/dev/null | grep -E "VGA|Display") gpu_info=$(lspci 2>/dev/null | grep -E "VGA|Display")
case "$gpu_info" in case "$gpu_info" in
*"Navi 1"*) *"Navi 1"*)
pyv="$(${python_cmd} -c 'import sys; print(".".join(map(str, sys.version_info[0:2])))')" export HSA_OVERRIDE_GFX_VERSION=10.3.0
if [[ $(bc <<< "$pyv <= 3.10") -eq 1 ]] if [[ -z "${TORCH_COMMAND}" ]]
then then
export HSA_OVERRIDE_GFX_VERSION=10.3.0 pyv="$(${python_cmd} -c 'import sys; print(".".join(map(str, sys.version_info[0:2])))')"
# Navi users will still use torch 1.13 because 2.0 does not seem to work. if [[ $(bc <<< "$pyv <= 3.10") -eq 1 ]]
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" then
else # Navi users will still use torch 1.13 because 2.0 does not seem to work.
printf "\e[1m\e[31mERROR: RX 5000 series GPUs must be using at max python 3.10, aborting...\e[0m" 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"
exit 1 else
printf "\e[1m\e[31mERROR: RX 5000 series GPUs must be using at max python 3.10, aborting...\e[0m"
exit 1
fi
fi fi
;; ;;
*"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0 *"Navi 2"*) export HSA_OVERRIDE_GFX_VERSION=10.3.0