【深度学习】训练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
1
2

构建环境：

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
1
2
3
4
5

安装完毕：

配置加速方案：

accelerate config
1

运行：

./gui.sh --share
1

docker run --net host --gpus device=0 -it kevinchina/deeplearning:train_kohya_ss bash