amd-stable-diffusion-webui-docker/Dockerfile at main · WaterBread/amd-stable-diffusion-webui-docker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
FROM alpine/git:latest AS builder
WORKDIR /opt
RUN git clone --depth 1 https://github.com/basujindal/stable-diffusion
RUN mkdir -p /opt/stable-diffusion/models/ldm/stable-diffusion-v1/
RUN git clone --depth 1 https://github.com/AUTOMATIC1111/stable-diffusion-webui
ADD sd-v1-4.ckpt /opt/stable-diffusion-webui/model.ckpt

FROM rocm/pytorch:rocm5.2_ubuntu20.04_py3.7_pytorch_1.11.0_navi21
COPY --from=builder /opt/stable-diffusion-webui /opt/stable-diffusion-webui
COPY --from=builder /opt/stable-diffusion /opt/stable-diffusion-webui/repositories/stable-diffusion

WORKDIR /opt/stable-diffusion-webui/repositories/stable-diffusion
RUN conda init
RUN conda env create -f environment.yaml
ENV PATH /opt/conda/envs/ldm/bin:$PATH
RUN echo "source activate ldm" >> ~/.bashrc

WORKDIR /opt/stable-diffusion-webui
RUN pip3 install -r requirements.txt

RUN pip3 install --upgrade torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1