Dockerfile

FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.5.1-py3.10-cuda11.8-ubuntu22.04-devel as base
ARG IMAGE=ruyi-mini-7b
ARG IMAGE_UPPER=Ruyi-Models
ARG BRANCH=gpu
RUN cd /root && git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git
WORKDIR /root/$IMAGE/$IMAGE_UPPER
RUN pip install -r requirements.txt


#########
# Prod  #
#########
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.5.1-py3.10-cuda11.8-ubuntu22.04-devel
ARG IMAGE=ruyi-mini-7b
ARG IMAGE_UPPER=Ruyi-Models
COPY chenyh/$IMAGE/frpc_linux_amd64_* /opt/conda/lib/python3.10/site-packages/gradio/
RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_*
COPY chenyh/$IMAGE/ /root/$IMAGE_UPPER/models
RUN apt-get update && apt-get install -y libgl1-mesa-glx libglib2.0-0 libsm6 libxrender1 libxext6
COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
COPY --from=base /root/$IMAGE/$IMAGE_UPPER /root/$IMAGE_UPPER
COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE/start.sh  /root/
COPY --from=base /root/$IMAGE/assets/  /root/assets/