Skip to content

Add creating global metrics recorder in file (#580) #105

Add creating global metrics recorder in file (#580)

Add creating global metrics recorder in file (#580) #105

name: Build and push base docker image (CUDA, NCCL)
on:
push:
paths:
- Dockerfile.base
- .github/workflows/build-and-push-base.yaml
workflow_dispatch:
env:
REGISTRY: ghcr.io
IMAGE_NAME: ${{ github.repository }}-base
CUDA_VERSION: 12_2
NCCL_VERSION: 2_22_3_1
jobs:
docker:
runs-on:
labels: ubuntu-22.04-64core
permissions:
packages: write
contents: read
attestations: write
id-token: write
steps:
- name: Checkout
uses: actions/checkout@v4
- name: Set up QEMU
uses: docker/setup-qemu-action@v3
- name: Set up Docker Buildx
uses: docker/setup-buildx-action@v3
- name: Log in to the Container registry
uses: docker/login-action@v3
with:
registry: ${{ env.REGISTRY }}
username: ${{ github.repository_owner }}
password: ${{ secrets.GITHUB_TOKEN }}
- name: Build and Push
uses: docker/build-push-action@v6
with:
context: .
push: true
tags: |
${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}:cuda${{ env.CUDA_VERSION }}-nccl${{ env.NCCL_VERSION }}
platforms: linux/amd64
cache-from: type=gha
cache-to: type=gha,mode=max
file: Dockerfile.base