|
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/11.7.1/ubuntu2204/devel/cudnn8/Dockerfile |
|
FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04 |
|
ENV DEBIAN_FRONTEND noninteractive |
|
|
|
WORKDIR /content |
|
|
|
RUN apt-get update -y && apt-get upgrade -y && apt-get install -y libgl1 libglib2.0-0 wget git git-lfs python3-pip python-is-python3 && pip3 install --upgrade pip |
|
|
|
RUN git clone -b v1.6 https://github.com/camenduru/stable-diffusion-webui && cd stable-diffusion-webui && sed -i -e 's/ start()/ #start()/g' launch.py && python launch.py --skip-torch-cuda-test |
|
|
|
COPY xformers.py /content/xformers.py |
|
|
|
RUN chmod -R 777 /content |
|
|
|
EXPOSE 7860 |
|
|
|
CMD python xformers.py |