mkdir -p /MY-DATA-DIR && cd /MY-DATA-DIR
mkdir models outputs
sudo chown 10000:$UID -R models outputs
sudo chmod 775 -R models outputs
docker run -it --name sdw --gpus all --network host \
-v $(pwd)/models:/app/stable-diffusion-webui/models \
-v $(pwd)/outputs:/app/stable-diffusion-webui/outputs \
--rm siutin/stable-diffusion-webui-docker:latest-cuda \
bash webui.sh --share
docker run -it --name sdw --network host \
-v $(pwd)/models:/app/stable-diffusion-webui/models \
-v $(pwd)/outputs:/app/stable-diffusion-webui/outputs \
--rm siutin/stable-diffusion-webui-docker:latest-cpu \
bash webui.sh --skip-torch-cuda-test --use-cpu all --share
# Nvidia CUDA image
nvidia-docker buildx build -f Dockerfile.cuda \
--platform linux/amd64 \
--build-arg BUILD_DATE=$(date -u +'%Y-%m-%dT%H:%M:%SZ') \
--build-arg BUILD_VERSION=custom-cuda \
-t siutin/stable-diffusion-webui-docker:custom-cuda .
# CPU only image
docker buildx build -f Dockerfile.cpu \
--platform linux/arm64 \
--build-arg BUILD_DATE=$(date -u +'%Y-%m-%dT%H:%M:%SZ') \
--build-arg BUILD_VERSION=custom-cpu \
-t siutin/stable-diffusion-webui-docker:custom-cpu .