Dockerfile 1.0 KB

123456789101112131415161718192021222324252627
  1. # Use the official vllm image for gpu with Ampere architecture and above (Compute Capability>=8.0)
  2. # Compute Capability version query (https://developer.nvidia.com/cuda-gpus)
  3. FROM vllm/vllm-openai:v0.10.1.1
  4. # Use the official vllm image for gpu with Turing architecture and below (Compute Capability<8.0)
  5. # FROM vllm/vllm-openai:v0.10.2
  6. # Install libgl for opencv support & Noto fonts for Chinese characters
  7. RUN apt-get update && \
  8. apt-get install -y \
  9. fonts-noto-core \
  10. fonts-noto-cjk \
  11. fontconfig \
  12. libgl1 && \
  13. fc-cache -fv && \
  14. apt-get clean && \
  15. rm -rf /var/lib/apt/lists/*
  16. # Install mineru latest
  17. RUN python3 -m pip install -U 'mineru[core]' --break-system-packages && \
  18. python3 -m pip cache purge
  19. # Download models and update the configuration file
  20. RUN /bin/bash -c "mineru-models-download -s huggingface -m all"
  21. # Set the entry point to activate the virtual environment and run the command line tool
  22. ENTRYPOINT ["/bin/bash", "-c", "export MINERU_MODEL_SOURCE=local && exec \"$@\"", "--"]