FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base
COPY Kolors /root/Kolors/
WORKDIR /root/Kolors
RUN conda create --name kolors -y python=3.8 && source activate kolors && 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
RUN conda create --name kolors python=3.8
COPY --from=base /opt/conda/envs/kolors/lib/python3.8/site-packages /opt/conda/envs/kolors/lib/python3.8/site-packages
COPY --from=base /root/Kolors /root/Kolors
COPY 启动器.ipynb run.sh /root/