"src/targets/gpu/device/multinomial.cpp" did not exist on "ccd08b4cca3b51380ec7176b25c8d5b818ffba3e"
Commit 469d1027 authored by chenpangpang's avatar chenpangpang
Browse files

修改Dockerfile适配dcu,修改启动器.ipynb增加适配硬件信息

parent e035a36b
FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel as base
FROM image.sourcefind.cn:5000/dcu/admin/base/jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel as base
ARG IMAGE=minicpm-v-2_6
ARG IMAGE_UPPER=MiniCPM-V-2_6
ARG BRANCH=dcu
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
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
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-cp310-cp310-manylinux_2_28_x86_64.whl'
RUN pip install flash_attn-*.whl && rm flash_attn-*.whl
#########
# Prod #
#########
FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel
FROM image.sourcefind.cn:5000/dcu/admin/base/jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel
ARG IMAGE=minicpm-v-2_6
ARG IMAGE_UPPER=MiniCPM-V-2_6
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
COPY chenyh/$IMAGE/frpc_linux_amd64_v0.2 /opt/conda/lib/python3.10/site-packages/gradio/
RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.2
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 /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
COPY --from=base /root/$IMAGE/assets/ /root/assets/
\ No newline at end of file
......@@ -10,6 +10,8 @@
"## 项目介绍\n",
"- 原项目地址:https://huggingface.co/spaces/openbmb/MiniCPM-V-2_6\n",
"- MiniCPM-V-2_6:一款聊天机器人,支持单图、多图、视频聊天,支持上下文小样本学习\n",
"## 运行资源\n",
"- 项目在L20显卡,cuda12.2上进行适配\n",
"## 使用说明\n",
"- 启动和重启 Notebook 点上方工具栏中的「重启并运行所有单元格」。出现如下内容就算成功了:\n",
" - `Running on local URL: http://0.0.0.0:7860`\n",
......@@ -53,7 +55,8 @@
"metadata": {
"collapsed": false
},
"id": "6dc59fbbcf222b6b"
"id": "6dc59fbbcf222b6b",
"execution_count": null
}
],
"metadata": {
......
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