gvisor.dev/gvisor@v0.0.0-20240520182842-f9d4d51c7e0f/images/gpu/stable-diffusion-xl/Dockerfile.x86_64 (about)

     1  FROM nvidia/cuda:12.3.1-devel-ubuntu22.04
     2  
     3  RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install --yes \
     4        python3 \
     5        python3-distutils \
     6        python3-pip \
     7        clang \
     8        wget \
     9        vim \
    10        git \
    11        libgl1 \
    12        libglib2.0-0 \
    13        libgl1-mesa-glx \
    14        golang
    15  
    16  RUN python3 -m pip install --ignore-installed \
    17        diffusers \
    18        transformers \
    19        accelerate \
    20        xformers \
    21        invisible-watermark
    22  
    23  RUN go install \
    24        github.com/TheZoraiz/ascii-image-converter@d05a757c5e02ab23e97b6f6fca4e1fbeb10ab559 && \
    25        mv "$HOME/go/bin/ascii-image-converter" /usr/bin/
    26  
    27  COPY download_checkpoints.py /tmp
    28  RUN chmod +x /tmp/download_checkpoints.py && \
    29        /tmp/download_checkpoints.py && \
    30        rm /tmp/download_checkpoints.py
    31  
    32  COPY generate_image generate_image.py /
    33  RUN chmod 555 /generate_image /generate_image.py
    34  ENV PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
    35  ENTRYPOINT ["/generate_image"]