-
Notifications
You must be signed in to change notification settings - Fork 3
174 lines (154 loc) · 6.03 KB
/
docker-publish.yml
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
name: Docker
# 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:
release:
types: [ "published" ]
pull_request:
branches: [ "main" ]
env:
# Use docker.io for Docker Hub if empty
REGISTRY: ghcr.io
JAX_VERSION : 0.4.23
JAX_CONDA_IMAGE_NAME : "yfukai/conda-jax"
# github.repository as <account>/<repo>
IMAGE_NAME: ${{ github.repository }}
jobs:
build:
runs-on: ubuntu-latest
permissions:
contents: read
packages: write
# This is used to complete the identity challenge
# with sigstore/fulcio when running outside of PRs.
id-token: write
steps:
- name: Checkout repository
uses: actions/checkout@v3
# Install the cosign tool except on PR
# https://github.com/sigstore/cosign-installer
- name: Install Cosign
# if: github.event_name != 'pull_request'
uses: sigstore/[email protected]
with:
cosign-release: 'v2.2.1'
# Workaround: https://github.com/docker/build-push-action/issues/461
- name: Setup Docker buildx
uses: docker/setup-buildx-action@79abd3f86f79a9d68a23c75a09a9a85889262adf
# Login against a Docker registry except on PR
# https://github.com/docker/login-action
- name: Log into registry ${{ env.REGISTRY }}
# if: github.event_name != 'pull_request'
uses: docker/login-action@28218f9b04b4f3f62068d7b6ce6ca5b26e35336c
with:
registry: ${{ env.REGISTRY }}
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}
# Extract metadata (tags, labels) for Docker
# https://github.com/docker/metadata-action
- name: Extract Docker metadata (JAX conda CPU)
id: meta_jax_conda
uses: docker/metadata-action@98669ae865ea3cffbcbaa878cf57c20bbf1c6c38
with:
images: ${{ env.REGISTRY }}/${{ env.JAX_CONDA_IMAGE_NAME }}
tags: |
type=schedule
type=ref,event=branch
type=ref,event=tag
type=ref,event=pr
- name: Extract first tag (JAX conda CPU)
id: extract_first_tag_cpu
run: |
# Extracting the first component
first_component=$(echo "${{ steps.meta_jax_conda.outputs.tags }}" | cut -d ',' -f 1)
# Setting the output
echo "meta_jax_conda_first_tag=$first_component" >> $GITHUB_OUTPUT
- name: Extract Docker metadata (JAX conda GPU)
id: meta_jax_conda_cuda
uses: docker/metadata-action@98669ae865ea3cffbcbaa878cf57c20bbf1c6c38
with:
images: ${{ env.REGISTRY }}/${{ env.JAX_CONDA_IMAGE_NAME }}
tags: |
type=schedule,suffix=-cuda
type=ref,event=branch,suffix=-cuda
type=ref,event=tag,suffix=-cuda
type=ref,event=pr,suffix=-cuda
- name: Extract first tag (JAX conda GPU)
id: extract_first_tag_gpu
run: |
# Extracting the first component
first_component=$(echo "${{ steps.meta_jax_conda_cuda.outputs.tags }}" | cut -d ',' -f 1)
# Setting the output
echo "meta_jax_conda_cuda_first_tag=$first_component" >> $GITHUB_OUTPUT
- name: Extract Docker metadata (CPU)
id: meta
uses: docker/metadata-action@98669ae865ea3cffbcbaa878cf57c20bbf1c6c38
with:
images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
tags: |
type=schedule
type=ref,event=branch
type=ref,event=tag
type=ref,event=pr
- name: Extract Docker metadata (GPU)
id: meta_cuda
uses: docker/metadata-action@98669ae865ea3cffbcbaa878cf57c20bbf1c6c38
with:
images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
tags: |
type=schedule,suffix=-cuda
type=ref,event=branch,suffix=-cuda
type=ref,event=tag,suffix=-cuda
type=ref,event=pr,suffix=-cuda
- name: Set up QEMU
uses: docker/setup-qemu-action@v2
# Build and push Docker image with Buildx
# https://github.com/docker/build-push-action
- name: Build and push Docker image (JAX conda CPU)
id: build-and-push-jax-conda-cpu
uses: docker/build-push-action@v5
with:
context: jax_image/
platforms: linux/amd64,linux/arm64/v8
push: true
tags: ${{ steps.meta_jax_conda.outputs.tags }}
labels: ${{ steps.meta_jax_conda.outputs.labels }}
build-args: |
JAX_VERSION=${{ env.JAX_VERSION }}
- name: Build and push Docker image (JAX conda GPU)
id: build-and-push-jax-conda-gpu
uses: docker/build-push-action@v5
with:
context: jax_image/
platforms: linux/amd64
push: true
tags: ${{ steps.meta_jax_conda_cuda.outputs.tags }}
labels: ${{ steps.meta_jax_conda_cuda.outputs.labels }}
build-args: |
JAX_VERSION=${{ env.JAX_VERSION }}
JAX_VERSION_EXTRA=cuda11_local
BASE_IMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
- name: Build and push Docker image (CPU)
id: build-and-push-cpu
uses: docker/build-push-action@v5
with:
context: .
platforms: linux/amd64,linux/arm64/v8
push: true
tags: ${{ steps.meta.outputs.tags }}
labels: ${{ steps.meta.outputs.labels }}
build-args: |
BASE_IMAGE=${{ env.REGISTRY }}/${{ env.JAX_CONDA_IMAGE_NAME }}:${{ env.meta_jax_conda_first_tag }}
- name: Build and push Docker image (GPU)
id: build-and-push-gpu
uses: docker/build-push-action@v5
with:
context: .
platforms: linux/amd64
push: true
tags: ${{ steps.meta_cuda.outputs.tags }}
labels: ${{ steps.meta_cuda.outputs.labels }}
build-args: |
BASE_IMAGE=${{ env.REGISTRY }}/${{ env.JAX_CONDA_IMAGE_NAME }}:${{ env.meta_jax_conda_cuda_first_tag }}