Browse Source

fix: update ppu.Dockerfile to include vLLM base image and specific package versions

myhloli 1 week ago
parent
commit
cd1c5c5e50
1 changed files with 3 additions and 0 deletions
  1. 3 0
      docker/china/ppu.Dockerfile

+ 3 - 0
docker/china/ppu.Dockerfile

@@ -1,5 +1,7 @@
 # Base image containing the LMDeploy inference environment, requiring amd64(x86-64) CPU + t-head PPU.
 # Base image containing the LMDeploy inference environment, requiring amd64(x86-64) CPU + t-head PPU.
 FROM crpi-4crprmm5baj1v8iv.cn-hangzhou.personal.cr.aliyuncs.com/lmdeploy_dlinfer/ppu:mineru-ppu
 FROM crpi-4crprmm5baj1v8iv.cn-hangzhou.personal.cr.aliyuncs.com/lmdeploy_dlinfer/ppu:mineru-ppu
+# Base image containing the vLLM inference environment, requiring amd64(x86-64) CPU + t-head PPU.
+# FROM crpi-vofi3w62lkohhxsp.cn-shanghai.personal.cr.aliyuncs.com/opendatalab-mineru/ppu:ppu-pytorch2.6.0-ubuntu24.04-cuda12.6-vllm0.8.5-py312
 
 
 # Install libgl for opencv support & Noto fonts for Chinese characters
 # Install libgl for opencv support & Noto fonts for Chinese characters
 RUN apt-get update && \
 RUN apt-get update && \
@@ -14,6 +16,7 @@ RUN apt-get update && \
 
 
 # Install mineru latest
 # Install mineru latest
 RUN python3 -m pip install -U 'mineru[core]' -i https://mirrors.aliyun.com/pypi/simple --break-system-packages && \
 RUN python3 -m pip install -U 'mineru[core]' -i https://mirrors.aliyun.com/pypi/simple --break-system-packages && \
+    python3 -m pip install numpy==1.26.4 opencv-python==4.11.0.86 huggingface_hub==0.36.0 -i https://mirrors.aliyun.com/pypi/simple --break-system-packages && \
     python3 -m pip cache purge
     python3 -m pip cache purge
 
 
 # Download models and update the configuration file
 # Download models and update the configuration file