mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-31 14:40:15 +01:00
90 lines
3.3 KiB
Bash
90 lines
3.3 KiB
Bash
#!/bin/bash
|
|
|
|
# detect if build-essential is missing or broken
|
|
if ! dpkg-query -W -f'${Status}' "build-essential" 2>/dev/null | grep -q "ok installed"; then
|
|
echo "build-essential not found or broken!
|
|
|
|
A C++ compiler is required to build needed Python packages!
|
|
To install one, run cmd_wsl.bat and enter these commands:
|
|
|
|
sudo apt-get update
|
|
sudo apt-get install build-essential
|
|
"
|
|
read -n1 -p "Continue the installer anyway? [y,n]" EXIT_PROMPT
|
|
# only continue if user inputs 'y' else exit
|
|
if ! [[ $EXIT_PROMPT == "Y" || $EXIT_PROMPT == "y" ]]; then exit; fi
|
|
fi
|
|
|
|
# deactivate any currently active conda env
|
|
conda deactivate 2> /dev/null
|
|
|
|
# config unlike other scripts, can't use current directory due to file IO bug in WSL, needs to be in virtual drive
|
|
INSTALL_DIR="$HOME/text-gen-install"
|
|
CONDA_ROOT_PREFIX="$INSTALL_DIR/installer_files/conda"
|
|
INSTALL_ENV_DIR="$INSTALL_DIR/installer_files/env"
|
|
MINICONDA_DOWNLOAD_URL="https://repo.anaconda.com/miniconda/Miniconda3-py310_23.1.0-1-Linux-x86_64.sh"
|
|
conda_exists="F"
|
|
|
|
# environment isolation
|
|
export PYTHONNOUSERSITE=1
|
|
unset PYTHONPATH
|
|
unset PYTHONHOME
|
|
export CUDA_PATH="$INSTALL_ENV_DIR"
|
|
export CUDA_HOME="$CUDA_PATH"
|
|
|
|
# /usr/lib/wsl/lib needs to be added to LD_LIBRARY_PATH to fix years-old bug in WSL where GPU drivers aren't linked properly
|
|
export LD_LIBRARY_PATH="$CUDA_HOME/lib:/usr/lib/wsl/lib:$LD_LIBRARY_PATH"
|
|
|
|
# open bash cli if called with 'wsl.sh cmd' with workarounds for existing conda
|
|
if [ "$1" == "cmd" ]; then
|
|
exec bash --init-file <(echo ". ~/.bashrc; conda deactivate 2> /dev/null; cd $INSTALL_DIR || cd $HOME; source $CONDA_ROOT_PREFIX/etc/profile.d/conda.sh; conda activate $INSTALL_ENV_DIR")
|
|
exit
|
|
fi
|
|
|
|
if [[ "$INSTALL_DIR" =~ " " ]]; then echo This script relies on Miniconda which can not be silently installed under a path with spaces. && exit; fi
|
|
|
|
# create install dir if missing and copy webui.py to install dir to maintain functionality without edit
|
|
if [ ! -d "$INSTALL_DIR" ]; then mkdir -p "$INSTALL_DIR" || exit; fi
|
|
cp -u "./webui.py" "$INSTALL_DIR"
|
|
|
|
# figure out whether git and conda needs to be installed
|
|
if "$CONDA_ROOT_PREFIX/bin/conda" --version &>/dev/null; then conda_exists="T"; fi
|
|
|
|
# (if necessary) install git and conda into a contained environment
|
|
# download miniconda
|
|
if [ "$conda_exists" == "F" ]; then
|
|
echo "Downloading Miniconda from $MINICONDA_DOWNLOAD_URL to $INSTALL_DIR/miniconda_installer.sh"
|
|
|
|
curl -Lk "$MINICONDA_DOWNLOAD_URL" > "$INSTALL_DIR/miniconda_installer.sh"
|
|
|
|
chmod u+x "$INSTALL_DIR/miniconda_installer.sh"
|
|
bash "$INSTALL_DIR/miniconda_installer.sh" -b -p $CONDA_ROOT_PREFIX
|
|
|
|
# test the conda binary
|
|
echo "Miniconda version:"
|
|
"$CONDA_ROOT_PREFIX/bin/conda" --version
|
|
fi
|
|
|
|
cd $INSTALL_DIR
|
|
|
|
# create the installer env
|
|
if [ ! -e "$INSTALL_ENV_DIR" ]; then
|
|
"$CONDA_ROOT_PREFIX/bin/conda" create -y -k --prefix "$INSTALL_ENV_DIR" python=3.10
|
|
fi
|
|
|
|
# check if conda environment was actually created
|
|
if [ ! -e "$INSTALL_ENV_DIR/bin/python" ]; then
|
|
echo "Conda environment is empty."
|
|
exit
|
|
fi
|
|
|
|
# activate installer env
|
|
source "$CONDA_ROOT_PREFIX/etc/profile.d/conda.sh" # otherwise conda complains about 'shell not initialized' (needed when running in a script)
|
|
conda activate "$INSTALL_ENV_DIR"
|
|
|
|
# setup installer env update env if called with 'wsl.sh update'
|
|
case "$1" in
|
|
("update") python webui.py --update;;
|
|
(*) python webui.py;;
|
|
esac
|