Create Dockerfile

Dockerfile

@@ -0,0 +1,24 @@
+# Dockerfile Public A10G
+
+# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.2.0/ubuntu2204/base/Dockerfile
+FROM nvidia/cuda:12.2.0-base-ubuntu22.04
+ENV DEBIAN_FRONTEND noninteractive
+
+RUN apt-get update -y \
+	apt-get install -y aria2 libgl1 libglib2.0-0 wget git git-lfs python3-pip python-is-python3 \
+	pip install -q torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2+cu118 torchtext==0.15.2 torchdata==0.6.1 --extra-index-url https://download.pytorch.org/whl/cu118 \
+	pip install xformers==0.0.20 triton==2.0.0 \
+	adduser --disabled-password --gecos '' user \
+	mkdir /content \
+	chown -R user:user /content \
+ 	git clone -b v2.6 https://github.com/camenduru/stable-diffusion-webui /content/stable-diffusion-webui \
+	sed -i -e 's/    start()/    #start()/g' /content/stable-diffusion-webui/launch.py \
+	python /content/stable-diffusion-webui/launch.py --skip-torch-cuda-test \
+	git -C /content/stable-diffusion-webui reset --hard \
+	wget https://github.com/camenduru/webui-docker/raw/main/header_patch.py -O /content/header_patch.py \
+	sed -i -e '/demo:/r /content/header_patch.py' /content/stable-diffusion-webui/modules/ui.py \
+	aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/juggernaut-xl/resolve/main/juggernautXL_version2.safetensors -d /content/stable-diffusion-webui/models/Stable-diffusion -o juggernautXL_version2.safetensors
+
+WORKDIR /content
+USER user
+CMD python /content/stable-diffusion-webui/launch.py --xformers --listen