5af8b0b921
Ubuntu 22.04's protobuf-compiler (v3.12) lacks proto3 optional field support. Install protoc v25.1 from GitHub releases instead, which matches the protoc version available in Debian bookworm-based images used by the other 5 services. Validated: docker build completes successfully, binary starts with ml_training_service --help. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
132 lines
4.5 KiB
Docker
132 lines
4.5 KiB
Docker
# Multi-stage build for Foxhunt ML Training Service
|
|
# Wave 112 Agent 18: CUDA-enabled production Docker image
|
|
|
|
# =============================================================================
|
|
# Builder Stage with CUDA Support
|
|
# =============================================================================
|
|
FROM nvidia/cuda:12.3.0-devel-ubuntu22.04 AS builder
|
|
|
|
# Install Rust
|
|
RUN apt-get update && apt-get install -y \
|
|
curl \
|
|
build-essential \
|
|
&& curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y \
|
|
&& . $HOME/.cargo/env \
|
|
&& rustup default stable \
|
|
&& rustup component add rustfmt clippy
|
|
|
|
ENV PATH="/root/.cargo/bin:${PATH}"
|
|
|
|
# Install build dependencies including CUDA toolkit
|
|
# Note: protobuf-compiler from Ubuntu 22.04 is v3.12 which lacks proto3 optional support.
|
|
# Install protoc v25 from GitHub releases instead.
|
|
RUN apt-get update && apt-get install -y \
|
|
pkg-config \
|
|
libssl-dev \
|
|
nvidia-cuda-toolkit \
|
|
unzip \
|
|
&& rm -rf /var/lib/apt/lists/* \
|
|
&& curl -sSL https://github.com/protocolbuffers/protobuf/releases/download/v25.1/protoc-25.1-linux-x86_64.zip \
|
|
-o /tmp/protoc.zip \
|
|
&& unzip -q /tmp/protoc.zip -d /usr/local bin/protoc 'include/*' \
|
|
&& rm /tmp/protoc.zip \
|
|
&& chmod +x /usr/local/bin/protoc
|
|
|
|
# Set working directory
|
|
WORKDIR /build
|
|
|
|
# Set CUDA environment variables to skip GPU detection during build
|
|
# CUDA 13.0 matches development environment, RTX 3050 Ti = compute capability 8.6
|
|
ENV CUDARC_CUDA_VERSION=13000
|
|
ENV CUDA_COMPUTE_CAP=86
|
|
|
|
# Copy workspace manifests
|
|
COPY Cargo.toml Cargo.lock ./
|
|
|
|
# Copy sqlx offline cache for compile-time query verification
|
|
COPY .sqlx ./.sqlx
|
|
|
|
# Copy workspace members to satisfy manifest dependencies
|
|
COPY trading_engine ./trading_engine
|
|
COPY risk ./risk
|
|
COPY risk-data ./risk-data
|
|
COPY trading-data ./trading-data
|
|
COPY tli ./tli
|
|
COPY ml ./ml
|
|
COPY ml-data ./ml-data
|
|
COPY data ./data
|
|
COPY backtesting ./backtesting
|
|
COPY adaptive-strategy ./adaptive-strategy
|
|
COPY common ./common
|
|
COPY storage ./storage
|
|
COPY model_loader ./model_loader
|
|
COPY market-data ./market-data
|
|
COPY database ./database
|
|
COPY config ./config
|
|
COPY web-gateway ./web-gateway
|
|
COPY ctrader-openapi ./ctrader-openapi
|
|
COPY foxhunt-deploy ./foxhunt-deploy
|
|
COPY services/backtesting_service ./services/backtesting_service
|
|
COPY services/broker_gateway_service ./services/broker_gateway_service
|
|
COPY services/trading_service ./services/trading_service
|
|
COPY services/ml_training_service ./services/ml_training_service
|
|
COPY services/api_gateway ./services/api_gateway
|
|
COPY services/load_tests ./services/load_tests
|
|
COPY services/stress_tests ./services/stress_tests
|
|
COPY services/integration_tests ./services/integration_tests
|
|
COPY services/data_acquisition_service ./services/data_acquisition_service
|
|
COPY services/trading_agent_service ./services/trading_agent_service
|
|
COPY tests ./tests
|
|
|
|
# Enable sqlx offline mode to use cached query metadata
|
|
ENV SQLX_OFFLINE=true
|
|
|
|
# Build the application (CUDA available at runtime via nvidia/cuda base image)
|
|
RUN cargo build --release -p ml_training_service
|
|
|
|
# =============================================================================
|
|
# Runtime Stage with CUDA Runtime
|
|
# =============================================================================
|
|
FROM nvidia/cuda:12.3.0-runtime-ubuntu22.04
|
|
|
|
# Install runtime dependencies
|
|
RUN apt-get update && apt-get install -y \
|
|
ca-certificates \
|
|
libssl3 \
|
|
curl \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Download and install grpc_health_probe
|
|
RUN curl -sSL https://github.com/grpc-ecosystem/grpc-health-probe/releases/download/v0.4.25/grpc_health_probe-linux-amd64 \
|
|
-o /usr/local/bin/grpc_health_probe && \
|
|
chmod +x /usr/local/bin/grpc_health_probe
|
|
|
|
# Create non-root user
|
|
RUN groupadd --system --gid 1000 foxhunt && \
|
|
useradd --system --uid 1000 --gid foxhunt --shell /bin/bash foxhunt
|
|
|
|
# Create application directories
|
|
RUN mkdir -p /app/config /app/logs /app/models /app/checkpoints && \
|
|
chown -R foxhunt:foxhunt /app
|
|
|
|
# Set working directory
|
|
WORKDIR /app
|
|
|
|
# Copy binary from builder
|
|
COPY --from=builder /build/target/release/ml_training_service ./ml_training_service
|
|
RUN chmod +x ./ml_training_service
|
|
|
|
# Switch to non-root user
|
|
USER foxhunt
|
|
|
|
# Expose gRPC, metrics, and health ports
|
|
EXPOSE 50053 9094 8080
|
|
|
|
# Health check using HTTP endpoint (no TLS required)
|
|
HEALTHCHECK --interval=10s --timeout=5s --start-period=30s --retries=3 \
|
|
CMD curl -f http://localhost:8080/health || exit 1
|
|
|
|
# Run the application with default serve command
|
|
ENTRYPOINT ["./ml_training_service"]
|
|
CMD ["serve"]
|