【深度学习】训练Stable Diffusion环境

仓库：
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