https://huggingface.co/spaces/Robert001/UniControl-Demo

FROM kevinchina/deeplearning:sd_base_cuda116clip4
RUN . ~/.bashrc && pip uninstall -y clip-interrogator accelerate transformers
RUN git clone https://huggingface.co/spaces/Robert001/UniControl-Demo
RUN rm /UniControl-Demo/app.py
RUN rm /UniControl-Demo/requirements.txt
COPY ./deps/app.py /UniControl-Demo/
COPY ./deps/requirements.txt /UniControl-Demo/
RUN . ~/.bashrc && cd /UniControl-Demo/ && conda activate sd-inf && pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple
RUN . ~/.bashrc && conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.6 -c pytorch -c nvidia -y
#RUN . ~/.bashrc && cd /UniControl-Demo/ && python app.py
RUN rm -rf /UniControl-Demo
RUN mkdir -p /UniControl-Demo
COPY ./UniControl-Demo/ /UniControl-Demo/
RUN mkdir -p "/root/.cache/huggingface/hub/models--openai--clip-vit-large-patch14/"
COPY "./models--openai--clip-vit-large-patch14/" "/root/.cache/huggingface/hub/models--openai--clip-vit-large-patch14/"
RUN mkdir -p "/root/.cache/torch/hub/checkpoints/"
COPY "./checkpoints/" "/root/.cache/torch/hub/checkpoints/"




WORKDIR /app
COPY ./deps/startProject.sh /app/
EXPOSE 7861
ENTRYPOINT ["bash", "/app/startProject.sh"]

docker push kevinchina/deeplearning:cuda116_unicontrol_base

docker push kevinchina/deeplearning:cuda116_unicontrol

运行：

docker run -d --gpus all -p 7861:7861 kevinchina/deeplearning:cuda116_unicontrol