Dockerfile 1.26 KB
Newer Older
chenpangpang's avatar
chenpangpang committed
1
FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel as base
chenpangpang's avatar
chenpangpang committed
2
3
ARG IMAGE=minicpm-v-2_6
ARG IMAGE_UPPER=MiniCPM-V-2_6
chenpangpang's avatar
chenpangpang committed
4
ARG BRANCH=dcu
chenpangpang's avatar
chenpangpang committed
5
6
7
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
chenpangpang's avatar
chenpangpang committed
8
9
10
RUN wget --no-check-certificate  --content-disposition \
    'https://cancon.hpccube.com:65024/directlink/4/flash_attn/DAS1.2/flash_attn-2.0.4+das.opt2.dtk24042-cp311-cp311-manylinux_2_28_x86_64.whl' && \
    pip install flash_attn-2.0.4+das.opt2.dtk24042-cp311-cp311-manylinux_2_28_x86_64.whl
chenpangpang's avatar
chenpangpang committed
11
12
13
14

#########
# Prod  #
#########
chenpangpang's avatar
chenpangpang committed
15
FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel
chenpangpang's avatar
chenpangpang committed
16
17
ARG IMAGE=minicpm-v-2_6
ARG IMAGE_UPPER=MiniCPM-V-2_6
chenpangpang's avatar
chenpangpang committed
18
19
COPY chenyh/$IMAGE/frpc_linux_amd64_v0.3 /opt/conda/lib/python3.10/site-packages/gradio/
RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.3
chenpangpang's avatar
chenpangpang committed
20
COPY chenyh/$IMAGE/openbmb/MiniCPM-V-2_6 /root/$IMAGE_UPPER/openbmb/MiniCPM-V-2_6
chenpangpang's avatar
chenpangpang committed
21
COPY --from=base /usr/local/lib/python3.10/dist-packages /usr/local/lib/python3.10/dist-packages
chenpangpang's avatar
chenpangpang committed
22
23
24
25
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