Dockerfile

FROM image.sourcefind.cn:5000/dcu/admin/base/jupyterlab-pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 as base
RUN cd /root && git clone -b dcu http://developer.hpccube.com/codes/chenpangpang/kolors.git
WORKDIR /root/kolors/Kolors
RUN pip install -r requirements.txt && python3 setup.py install
RUN rm -r build dist kolors.egg-info

#########
# Prod  #
#########
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04
COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
COPY --from=base /root/kolors/Kolors /root/Kolors
COPY --from=base /root/kolors/启动器.ipynb /root/kolors/run.sh /root/
RUN mkdir /root/Kolors/weight
# 模型仓库http://113.200.138.88:18080/aimodels/Kolors.git，放置在/public/home/openaimodels/dockerFileTemp/chenyh
COPY chenyh/Kolors /root/Kolors/weight/Kolors