ML Server Container Builder

VerifiedCaution

Build and deploy Docker containers for ML server components with multi-model support (FinRL, StockMixer, etc.) and cache/registry options.

Sby Skills Guide Bot
Data & AIIntermediate
206/2/2026
Claude Code
#ml#docker#build#containers#models

Recommended for

Our review

Builds Docker containers for ML server components using docker compose.

Strengths

  • Automates building multiple ML models together or individually.
  • Supports advanced options like push to registry, no-cache builds, and CPU-only builds.
  • Clear structure for different models (FinRL, StockMixer, etc.).

Limitations

  • Depends on having the appropriate Dockerfiles and docker compose setup in the project.
  • May fail if model weights need to be downloaded separately or if CUDA dependencies are missing.
  • Does not handle deployment or registry management beyond pushing.
When to use it

Use this skill to quickly build a Docker image for an ML server with multiple model components.

When not to use it

Do not use it if you need to deploy models to Kubernetes or manage complex CI/CD pipelines.

Security analysis

Caution
Quality score85/100

The skill uses Docker commands (build, push) which are powerful and could be misused if the user runs malicious content, but the instructions themselves are legitimate and do not include destructive, exfiltrating, or obfuscated actions. The skill is safe for its intended purpose of building ML containers.

No concerns found

Examples

Build all ML containers with push
/build-ml --push
Build FinRL model without cache
/build-ml --no-cache finrl
Build CPU-only containers
/build-ml --cpu

name: build-ml description: Build ML server Docker containers argument-hint: "[--push|--no-cache|model]"

ML Server Builder

Build Docker containers for ML server components.

Usage

  • /build-ml - Build all ML containers
  • /build-ml --no-cache - Build without cache
  • /build-ml --push - Build and push to registry
  • /build-ml finrl - Build specific model container
  • /build-ml --cpu - Build CPU-only versions

ML Server Structure

services/ml-server/
├── docker-compose.yml      # ML services orchestration
├── docker-compose.dev.yml  # Development config
├── Dockerfile              # Base ML image
├── finrl/                  # FinRL deep learning
├── stockmixer/             # Stock mixing models
├── master/                 # Master orchestrator
├── samba/                  # Samba models
├── macrohft/               # Macro HFT models
└── src/                    # Shared ML code

Available Models

  • finrl - Deep reinforcement learning (FinRL)
  • stockmixer - Multi-asset mixing
  • master - Model orchestration
  • samba - Samba-based models
  • macrohft - Macro HFT strategies

Instructions

When this skill is invoked:

  1. Parse arguments:

    • --no-cache: Add --no-cache to docker build
    • --push: Push to container registry after build
    • --cpu: Use CPU-only base images
    • Model name: Build only that model's container
  2. Navigate to ML server directory:

    cd services/ml-server
    
  3. Build containers:

    # All containers
    docker compose build $NO_CACHE
    
    # Specific model
    docker compose build $MODEL $NO_CACHE
    
  4. If --push specified:

    • Verify registry credentials
    • Tag images appropriately
    • Push to registry:
      docker compose push $MODEL
      
  5. Report build results:

    • Build time per container
    • Final image sizes
    • Any build warnings
    • Layer cache efficiency
  6. Common build issues:

    • CUDA not found: Use --cpu flag or install NVIDIA container toolkit
    • Out of memory: Increase Docker memory limit
    • Pip install fails: Check requirements.txt for version conflicts
    • Model weights missing: Download from model registry first

Image Tags

  • latest - Most recent build
  • v{version} - Release versions
  • {git-sha} - Specific commit builds
  • cpu - CPU-only builds
Related skills