Commit e9414793 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 解决依赖包版本问题

parent e9040775
......@@ -5,6 +5,10 @@ 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/onnxruntime/DAS1.2/onnxruntime-1.15.0+das.opt1.dtk24042-cp310-cp310-manylinux_2_28_x86_64.whl' && \
pip install onnxruntime-1.15.0+das.opt1.dtk24042-cp310-cp310-manylinux_2_28_x86_64.whl
RUN pip install numpy==1.24.4 'scipy<1.13.0' opencv-python==3.4.*
#########
# Prod #
......@@ -12,9 +16,8 @@ RUN pip install -r requirements.txt
FROM image.sourcefind.cn:5000/dcu/admin/base/jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel
ARG IMAGE=photomaker-v2
ARG IMAGE_UPPER=PhotoMaker-V2
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 chenyh/$IMAGE/frpc_linux_amd64_v* /opt/conda/lib/python3.10/site-packages/gradio/
RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v*
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/
......
......@@ -3,3 +3,4 @@ accelerate
omegaconf
peft
insightface==0.7.3
gradio
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