FROM debian:12.11
RUN /bin/rm -f /etc/apt/sources.list.d/debian.sources && \
echo "deb http://mirrors.nju.edu.cn/debian/ bookworm main contrib non-free" > /etc/apt/sources.list && \
echo "deb http://mirrors.nju.edu.cn/debian/ bookworm-updates main contrib non-free" >> /etc/apt/sources.list && \
echo "deb http://mirrors.nju.edu.cn/debian-security bookworm-security/updates main contrib non-free" >> /etc/apt/sources.list && \
apt-get update && \
apt-get install -y python3 python3-pip python3-venv && \
/bin/rm -rf /var/lib/apt/lists/* && \
/bin/mv /usr/lib/python3.11/EXTERNALLY-MANAGED /usr/lib/python3.11/EXTERNALLY-MANAGED.bak && \
pip install -i https://mirrors.huaweicloud.com/repository/pypi/simple --no-cache-dir flask && \
pip install -i https://mirrors.huaweicloud.com/repository/pypi/simple --no-cache-dir transformers && \
pip install -i https://mirrors.huaweicloud.com/repository/pypi/simple --no-cache-dir torch && \
pip install -i https://mirrors.huaweicloud.com/repository/pypi/simple --no-cache-dir sentencepiece && \
pip install -i https://mirrors.huaweicloud.com/repository/pypi/simple --no-cache-dir protobuf && \
pip install -i https://mirrors.huaweicloud.com/repository/pypi/simple --no-cache-dir accelerate
WORKDIR /app
services:
rerankmodel:
image: python-rerankmodel:0.1
container_name: rerankmodel
environment:
TZ: "Asia/Shanghai"
volumes:
- ./rerankmodel:/app:rw
ports:
- "5001:5001"
working_dir: /app
entrypoint: ["python3","-u","rerank.py"]
Categories:
docker与kubernetes