33 lines
1.3 KiB
Docker
33 lines
1.3 KiB
Docker
# Use DaoCloud mirrored vllm image for China region for gpu with Ampere architecture and above (Compute Capability>=8.0)
|
|
# Compute Capability version query (https://developer.nvidia.com/cuda-gpus)
|
|
FROM docker.m.daocloud.io/vllm/vllm-openai:v0.10.1.1
|
|
|
|
# Use the official vllm image
|
|
# FROM vllm/vllm-openai:v0.10.1.1
|
|
|
|
# Use DaoCloud mirrored vllm image for China region for gpu with Turing architecture and below (Compute Capability<8.0)
|
|
# FROM docker.m.daocloud.io/vllm/vllm-openai:v0.10.2
|
|
|
|
# Use the official vllm image
|
|
# FROM vllm/vllm-openai:v0.10.2
|
|
|
|
# Install libgl for opencv support & Noto fonts for Chinese characters
|
|
RUN apt-get update && \
|
|
apt-get install -y \
|
|
fonts-noto-core \
|
|
fonts-noto-cjk \
|
|
fontconfig \
|
|
libgl1 && \
|
|
fc-cache -fv && \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install mineru latest
|
|
RUN python3 -m pip install -U 'mineru[core]' -i https://mirrors.aliyun.com/pypi/simple --break-system-packages && \
|
|
python3 -m pip cache purge
|
|
|
|
# Download models and update the configuration file
|
|
RUN /bin/bash -c "mineru-models-download -s modelscope -m all"
|
|
|
|
# Set the entry point to activate the virtual environment and run the command line tool
|
|
ENTRYPOINT ["/bin/bash", "-c", "export MINERU_MODEL_SOURCE=local && exec \"$@\"", "--"] |