FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base
RUN cd /root && git clone -b gpu https://developer.hpccube.com/codes/chenpangpang/tile-upscaler
WORKDIR /root/tile-upscaler/Tile-Upscaler
RUN pip install -r requirements.txt
COPY chenyh/tile-upscaler/models /root/tile-upscaler/Tile-Upscaler/models

#########
# 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/tile-upscaler/Tile-Upscaler /root/Tile-Upscaler