Commit e035a36b authored by chenpangpang's avatar chenpangpang
Browse files

feat: dcu初始提交

parent b14040ec
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.1.2-py3.10-cuda11.8-ubuntu22.04 as base FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel as base
ARG IMAGE=minicpm-v-2_6 ARG IMAGE=minicpm-v-2_6
ARG IMAGE_UPPER=MiniCPM-V-2_6 ARG IMAGE_UPPER=MiniCPM-V-2_6
ARG BRANCH=gpu ARG BRANCH=dcu
RUN cd /root && git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git RUN cd /root && git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git
WORKDIR /root/$IMAGE/$IMAGE_UPPER WORKDIR /root/$IMAGE/$IMAGE_UPPER
RUN pip install -r requirements.txt RUN pip install -r requirements.txt
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
######### #########
# Prod # # Prod #
######### #########
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.1.2-py3.10-cuda11.8-ubuntu22.04 FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel
ARG IMAGE=minicpm-v-2_6 ARG IMAGE=minicpm-v-2_6
ARG IMAGE_UPPER=MiniCPM-V-2_6 ARG IMAGE_UPPER=MiniCPM-V-2_6
COPY chenyh/$IMAGE/frpc_linux_amd64_v0.2 /usr/local/lib/python3.10/dist-packages/gradio/ COPY chenyh/$IMAGE/frpc_linux_amd64_v0.3 /opt/conda/lib/python3.10/site-packages/gradio/
RUN chmod +x /usr/local/lib/python3.10/dist-packages/gradio/frpc_linux_amd64_v0.2 RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.3
COPY chenyh/$IMAGE/openbmb/MiniCPM-V-2_6 /root/$IMAGE_UPPER/openbmb/MiniCPM-V-2_6 COPY chenyh/$IMAGE/openbmb/MiniCPM-V-2_6 /root/$IMAGE_UPPER/openbmb/MiniCPM-V-2_6
COPY --from=base /usr/local/lib/python3.10/dist-packages /usr/local/lib/python3.10/dist-packages COPY --from=base /usr/local/lib/python3.10/dist-packages /usr/local/lib/python3.10/dist-packages
COPY --from=base /root/$IMAGE/$IMAGE_UPPER /root/$IMAGE_UPPER COPY --from=base /root/$IMAGE/$IMAGE_UPPER /root/$IMAGE_UPPER
......
Pillow
torch
torchvision
transformers
sentencepiece
https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.2/flash_attn-2.6.2+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
opencv-python
decord decord
#gradio==4.22.0 gradio
gradio==4.41.0
http://thunlp.oss-cn-qingdao.aliyuncs.com/multi_modal/never_delete/modelscope_studio-0.4.0.9-py3-none-any.whl http://thunlp.oss-cn-qingdao.aliyuncs.com/multi_modal/never_delete/modelscope_studio-0.4.0.9-py3-none-any.whl
accelerate accelerate
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment