fix on torch_command + tested on debian

This commit is contained in:
orionaskatu 2022-09-13 17:48:10 +02:00 committed by AUTOMATIC1111
parent 7bf76af40a
commit d62fbcc5aa

View File

@ -41,6 +41,12 @@ then
venv_dir="venv"
fi
# install command for torch
if [[ -z "${TORCH_COMMAND}" ]]
then
export TORCH_COMMAND=(python3 -m pip install torch==1.12.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113)
fi
# Do not reinstall existing pip packages on Debian/Ubuntu
export PIP_IGNORE_INSTALLED=0