base-images/images.yaml at main · devopshobbies/base-images · GitHub

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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
# =============================================================================
# Base Images Configuration
# =============================================================================
# This file defines all Docker base images, their versions, and build settings.
# GitHub Actions reads this file to generate build matrices dynamically.
# =============================================================================

# Global settings
settings:
  registry: docker.io
  organization: devopshobbies
  # Default platforms for multi-arch builds
  default_platforms:
    - linux/amd64
    - linux/arm64

# Image definitions
images:
  # ---------------------------------------------------------------------------
  # ML/PyTorch - Production-ready PyTorch images with CUDA support
  # ---------------------------------------------------------------------------
  pytorch:
    path: images/ml/pytorch
    description: "Production-ready PyTorch base image with CUDA support"
    # Version matrix - each combination creates a separate image tag
    # Essential combinations only (4 variants)
    versions:
      cuda:
        - "12.1"
      python:
        - "3.10"
        - "3.11"
      pytorch:
        - "2.2"
        - "2.4"
    # Default versions (used for 'latest' tag)
    defaults:
      cuda: "12.1"
      python: "3.11"
      pytorch: "2.4"
    # Build arguments mapping
    build_args:
      CUDA_VERSION: cuda
      PYTHON_VERSION: python
      PYTORCH_VERSION: pytorch
    # Platforms (overrides global default if specified)
    platforms:
      - linux/amd64
    # Tags to generate
    tags:
      # Pattern: {image}-cuda{cuda}-py{python}-torch{pytorch}
      pattern: "pytorch-cuda{cuda}-py{python}-torch{pytorch}"
      # Also tag as 'latest' when default versions are used
      include_latest: true
    # Testing configuration
    test:
      enabled: true
      script: "test.sh"
      timeout: 300

  # ---------------------------------------------------------------------------
  # ML/TensorFlow - Production-ready TensorFlow images with CUDA support
  # ---------------------------------------------------------------------------
  tensorflow:
    path: images/ml/tensorflow
    description: "Production-ready TensorFlow base image with CUDA support"
    # Version matrix - pin full semver (pip has no tensorflow==2.16).
    # Python limited to versions available on the Ubuntu 22.04 CUDA base.
    versions:
      cuda:
        - "12.1"
      python:
        - "3.10"
        - "3.11"
      tensorflow:
        - "2.16.2"
        - "2.17.1"
    # Default versions (used for 'latest' tag)
    defaults:
      cuda: "12.1"
      python: "3.11"
      tensorflow: "2.17.1"
    # Build arguments mapping
    build_args:
      CUDA_VERSION: cuda
      PYTHON_VERSION: python
      TENSORFLOW_VERSION: tensorflow
    # Platforms (overrides global default if specified)
    platforms:
      - linux/amd64
    # Tags to generate
    tags:
      # Pattern: tensorflow-cuda{cuda}-py{python}-tf{tensorflow}
      pattern: "tensorflow-cuda{cuda}-py{python}-tf{tensorflow}"
      # Also tag as 'latest' when default versions are used
      include_latest: true
    # Testing configuration
    test:
      enabled: true
      script: "test.sh"
      timeout: 300