File size: 794 Bytes
36c78b1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
FROM ubuntu:22.04
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && \
apt-get install -y python3.11 python3-pip python3.11-venv curl && \
apt-get clean && rm -rf /var/lib/apt/lists/*
WORKDIR /opt/bit_transformer
COPY . .
ARG TORCH_CUDA=cpu
RUN pip3 install --no-cache-dir --upgrade pip && \
if [ "$TORCH_CUDA" = "cu118" ]; then \
pip3 install torch==2.7.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118; \
else \
pip3 install torch==2.7.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu; \
fi && \
pip3 install -r requirements.txt
ENV MCP_SERVER_ADDR=http://127.0.0.1:7000
EXPOSE 5000 7000
RUN chmod +x start.sh
HEALTHCHECK CMD curl -f http://localhost:7000/health || exit 1
CMD ["/opt/bit_transformer/start.sh"]
|