启动容器:
docker run -it --rm --gpus all --net host --shm-size=16g -v /root/xiedong:/root/xiedong kevinchina/deeplearning:pytorch2.3.0-cuda12.1-cudnn8-devel-xformers-lora-train bash
启动webui:
cd /workspace/lora-scripts
python gui.py --host "0.0.0.0" --port 7874 --tensorboard-host "0.0.0.0" --tensorboard-port 6007
训练 Stable Diffusion XL (SDXL