Dockerfile

FROM cnstark/pytorch:2.0.1-py3.10.11-ubuntu22.04

WORKDIR /code

RUN apt-get update && apt-get install -y git
#RUN pip3 install torch==2.0.1 --index-url https://download.pytorch.org/whl/cpu
#RUN git clone https://github.com/rupeshs/fastsdcpu && chmod a+x ./fastsdcpu/install.sh && chmod a+x ./fastsdcpu/start-webui.sh && /code/fastsdcpu/install.sh

RUN git clone https://github.com/rupeshs/fastsdcpu

WORKDIR /code/fastsdcpu
RUN chmod a+x install.sh && chmod a+x start-webui.sh 
RUN /code/fastsdcpu/install.sh || { exit 0; }

RUN apt-get install -y sudo
RUN useradd -rm -d /home/hf -s /bin/bash -g root -G sudo -u 1001 hf
USER hf

CMD ["sudo", "./start-webui.sh"]