仓库:
https://github.com/bmaltais/kohya_ss.git
基础镜像:
from kevinchina/deeplearning:sdxllighting_trt_nginx_002api
docker run --net host --gpus device=0 -e APIWORKS=1 -it t1:t1 bash
构建环境:
sudo -i
git clone https://github.com/bmaltais/kohya_ss.git
sudo apt update -y && sudo apt install -y python3-tk
cd kohya_ss
./setup.sh -d /root/kohya_ss/ -vvv
安装完毕:
配置加速方案:
accelerate config
运行:
./gui.sh --share
docker run --net host --gpus device=0 -it kevinchina/deeplearning:train_kohya_ss bash