| 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"] | |