Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add: cuda gpu capable images #137

Draft
wants to merge 1 commit into
base: main
Choose a base branch
from
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
123 changes: 123 additions & 0 deletions .github/workflows/build-udi-cuda-11.8.0.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,123 @@
name: Publish Build 11.8.0

# This workflow uses actions that are not certified by GitHub.
# They are provided by a third-party and are governed by
# separate terms of service, privacy policy, and support
# documentation.

on: # yamllint disable-line rule:truthy
# schedule:
# - cron: '5 4 * * 6'
push:
branches: ["*"]
# Publish semver tags as releases.
# tags: ["v*.*.*"]
paths:
- ".github/workflows/**"
- "udi-cuda/ubi8/**"
- "python/ubi8/**"
pull_request:
branches: ["main"]
paths:
- "udi-cuda/ubi8/**"
- "python/ubi8/**"

env:
IMAGE_NAME: udi-cuda
IMAGE_VERSION: 11.8.0
GHCR_REGISTRY: ghcr.io
GHCR_USER: ${{ github.repository_owner }}
GHCR_IMAGE: ${GHCR_REGISTRY}/${GHCR_USER}/${IMAGE_NAME}


jobs:
publish:
runs-on: ubuntu-latest
permissions:
contents: read
packages: write

steps:
- name: Checkout repository
uses: actions/[email protected]

- name: Prepare
id: prep
run: |
VERSION=${{ env.IMAGE_VERSION }}
GHCR_IMAGE=${{ env.GHCR_IMAGE }}
MINOR=${VERSION%.*}
MAJOR=${VERSION%%.*}
TAGS="${GHCR_IMAGE}:${VERSION}"
echo ::set-output name=name::${{ env.IMAGE_NAME }}
echo ::set-output name=tags::${TAGS}
echo ::set-output name=version::${VERSION}
echo ::set-output name=ghcr_image::${GHCR_IMAGE}

# https://github.com/docker/login-action
- name: Log into GitHub Container Registry
if: github.event_name != 'pull_request'
uses: docker/[email protected]
with:
registry: ${{ env.GHCR_REGISTRY }}
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}

# https://github.com/docker/metadata-action
- name: Extract Docker metadata
id: meta
uses: docker/metadata-action@96383f45573cb7f253c731d3b3ab81c87ef81934 # v5.0.0
with:
images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}

# https://github.com/docker/build-push-action
- name: Build and push runtime
uses: docker/[email protected]
with:
context: containers/${{ steps.prep.outputs.name }}/ubi8
file: containers/${{ steps.prep.outputs.name }}/ubi8/Dockerfile
push: ${{ github.event_name != 'pull_request' }}
build-args: |
IMAGE_NAME=docker.io/nvidia/cuda:${{ steps.prep.outputs.version }}-cudnn8-runtime-ubi8
TARGETARCH=amd64
tags: ${{ steps.prep.outputs.tags }}-cudnn8-runtime-ubi8
labels: org.opencontainers.image.version=${{ steps.prep.outputs.version }}-cudnn8-runtime-ubi8

# https://github.com/docker/build-push-action
- name: Build and push devel
uses: docker/[email protected]
with:
context: containers/${{ steps.prep.outputs.name }}/ubi8
file: containers/${{ steps.prep.outputs.name }}/ubi8/Dockerfile
push: ${{ github.event_name != 'pull_request' }}
build-args: |
IMAGE_NAME=docker.io/nvidia/cuda:${{ steps.prep.outputs.version }}-cudnn8-devel-ubi8
TARGETARCH=amd64
tags: ${{ steps.prep.outputs.tags }}-cudnn8-devel-ubi8
labels: org.opencontainers.image.version=${{ steps.prep.outputs.version }}-cudnn8-devel-ubi8

# https://github.com/docker/build-push-action
- name: Build and push devel python 3.9
uses: docker/[email protected]
with:
context: python/ubi8/3.9
file: python/ubi8/3.9/Dockerfile
push: ${{ github.event_name != 'pull_request' }}
build-args: |
IMAGE_NAME=${{ steps.prep.outputs.ghcr_image }}:${{ steps.prep.outputs.version }}-cudnn8-devel-ubi8
TARGETARCH=amd64
tags: ${{ steps.prep.outputs.tags }}-cudnn8-devel-ubi8-python39,${{ steps.prep.outputs.ghcr_image }}:latest
labels: org.opencontainers.image.version=${{ steps.prep.outputs.version }}-cudnn8-devel-ubi8-python39

# https://github.com/docker/build-push-action
- name: Build and push devel python 3.11
uses: docker/[email protected]
with:
context: python/ubi8/3.11
file: python/ubi8/3.11/Dockerfile
push: ${{ github.event_name != 'pull_request' }}
build-args: |
IMAGE_NAME=${{ steps.prep.outputs.ghcr_image }}:${{ steps.prep.outputs.version }}-cudnn8-devel-ubi8
TARGETARCH=amd64
tags: ${{ steps.prep.outputs.tags }}-cudnn8-devel-ubi8-python311
labels: org.opencontainers.image.version=${{ steps.prep.outputs.version }}-cudnn8-devel-ubi8-python311
123 changes: 123 additions & 0 deletions .github/workflows/build-udi-cuda-12.2.0.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,123 @@
name: Publish Build 12.2.0

# This workflow uses actions that are not certified by GitHub.
# They are provided by a third-party and are governed by
# separate terms of service, privacy policy, and support
# documentation.

on: # yamllint disable-line rule:truthy
# schedule:
# - cron: '5 4 * * 6'
push:
branches: ["*"]
# Publish semver tags as releases.
# tags: ["v*.*.*"]
paths:
- ".github/workflows/**"
- "udi-cuda/ubi8/**"
- "python/ubi8/**"
pull_request:
branches: ["main"]
paths:
- "udi-cuda/ubi8/**"
- "python/ubi8/**"

env:
IMAGE_NAME: udi-cuda
IMAGE_VERSION: 12.2.0
GHCR_REGISTRY: ghcr.io
GHCR_USER: ${{ github.repository_owner }}
GHCR_IMAGE: ${GHCR_REGISTRY}/${GHCR_USER}/${IMAGE_NAME}


jobs:
publish:
runs-on: ubuntu-latest
permissions:
contents: read
packages: write

steps:
- name: Checkout repository
uses: actions/[email protected]

- name: Prepare
id: prep
run: |
VERSION=${{ env.IMAGE_VERSION }}
GHCR_IMAGE=${{ env.GHCR_IMAGE }}
MINOR=${VERSION%.*}
MAJOR=${VERSION%%.*}
TAGS="${GHCR_IMAGE}:${VERSION}"
echo ::set-output name=name::${{ env.IMAGE_NAME }}
echo ::set-output name=tags::${TAGS}
echo ::set-output name=version::${VERSION}
echo ::set-output name=ghcr_image::${GHCR_IMAGE}

# https://github.com/docker/login-action
- name: Log into GitHub Container Registry
if: github.event_name != 'pull_request'
uses: docker/[email protected]
with:
registry: ${{ env.GHCR_REGISTRY }}
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}

# https://github.com/docker/metadata-action
- name: Extract Docker metadata
id: meta
uses: docker/metadata-action@96383f45573cb7f253c731d3b3ab81c87ef81934 # v5.0.0
with:
images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}

# https://github.com/docker/build-push-action
- name: Build and push runtime
uses: docker/[email protected]
with:
context: containers/${{ steps.prep.outputs.name }}/ubi8
file: containers/${{ steps.prep.outputs.name }}/ubi8/Dockerfile
push: ${{ github.event_name != 'pull_request' }}
build-args: |
IMAGE_NAME=docker.io/nvidia/cuda:${{ steps.prep.outputs.version }}-runtime-ubi8
TARGETARCH=amd64
tags: ${{ steps.prep.outputs.tags }}-runtime-ubi8
labels: org.opencontainers.image.version=${{ steps.prep.outputs.version }}-runtime-ubi8

# https://github.com/docker/build-push-action
- name: Build and push devel
uses: docker/[email protected]
with:
context: containers/${{ steps.prep.outputs.name }}/ubi8
file: containers/${{ steps.prep.outputs.name }}/ubi8/Dockerfile
push: ${{ github.event_name != 'pull_request' }}
build-args: |
IMAGE_NAME=docker.io/nvidia/cuda:${{ steps.prep.outputs.version }}-devel-ubi8
TARGETARCH=amd64
tags: ${{ steps.prep.outputs.tags }}-devel-ubi8
labels: org.opencontainers.image.version=${{ steps.prep.outputs.version }}-devel-ubi8

# https://github.com/docker/build-push-action
- name: Build and push devel python 3.9
uses: docker/[email protected]
with:
context: python/ubi8/3.9
file: python/ubi8/3.9/Dockerfile
push: ${{ github.event_name != 'pull_request' }}
build-args: |
IMAGE_NAME=${{ steps.prep.outputs.ghcr_image }}:${{ steps.prep.outputs.version }}-devel-ubi8
TARGETARCH=amd64
tags: ${{ steps.prep.outputs.tags }}-devel-ubi8-python39
labels: org.opencontainers.image.version=${{ steps.prep.outputs.version }}-devel-ubi8-python39

# https://github.com/docker/build-push-action
- name: Build and push devel python 3.11
uses: docker/[email protected]
with:
context: python/ubi8/3.11
file: python/ubi8/3.11/Dockerfile
push: ${{ github.event_name != 'pull_request' }}
build-args: |
IMAGE_NAME=${{ steps.prep.outputs.ghcr_image }}:${{ steps.prep.outputs.version }}-devel-ubi8
TARGETARCH=amd64
tags: ${{ steps.prep.outputs.tags }}-devel-ubi8-python311
labels: org.opencontainers.image.version=${{ steps.prep.outputs.version }}-devel-ubi8-python311
Loading