#!/usr/bin/bash

script_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"

echo "Installing torch & xformers..."

    echo "install torch 2.0.1+cu117"
    pip -q install torch==2.0.1 torchvision xformers triton 

echo "Installing deps..."
cd "$script_dir/sd-scripts" || exit

pip install --upgrade --ignore-installed -r requirements.txt

pip install tensorboard
apt-get update
apt-get install sudo
pip install protobuf==3.20.3 
pip install --upgrade --no-deps pytorch-optimizer -i https://mirror.baidu.com/pypi/simple
DEBIAN_FRONTEND=noninteractive apt-get install -y tzdata
DEBIAN_FRONTEND=noninteractive apt-get install -y python3-tk
cd "$script_dir" || exit




echo "依赖安装完成请继续下一步"