Skip to content

Commit b6b4a9d

Browse files
authored
add aarch64 cuda12
1 parent 3b8a796 commit b6b4a9d

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

.github/workflows/docker.yml

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,18 @@ jobs:
214214
needs: x86_64-scipy
215215
if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }}
216216

217+
aarch64-tensorflow-cuda:
218+
uses: ./.github/workflows/docker-build-test-upload.yml
219+
with:
220+
parent-image: scipy-notebook
221+
image: tensorflow-notebook
222+
variant: cuda
223+
platform: aarch64
224+
runs-on: ubuntu-24.04-arm
225+
timeout-minutes: 25
226+
needs: aarch64-scipy
227+
if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }}
228+
217229
x86_64-tensorflow-cuda:
218230
uses: ./.github/workflows/docker-build-test-upload.yml
219231
with:
@@ -248,6 +260,18 @@ jobs:
248260
needs: x86_64-scipy
249261
if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }}
250262

263+
aarch64-pytorch-cuda12:
264+
uses: ./.github/workflows/docker-build-test-upload.yml
265+
with:
266+
parent-image: scipy-notebook
267+
image: pytorch-notebook
268+
variant: cuda12
269+
platform: aarch64
270+
runs-on: ubuntu-24.04
271+
timeout-minutes: 25
272+
needs: aarch64-scipy
273+
if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }}
274+
251275
x86_64-pytorch-cuda11:
252276
uses: ./.github/workflows/docker-build-test-upload.yml
253277
with:
@@ -392,7 +416,9 @@ jobs:
392416
aarch64-r,
393417
aarch64-julia,
394418
aarch64-tensorflow,
419+
aarch64-tensorflow-cuda,
395420
aarch64-pytorch,
421+
aarch64-pytorch-cuda12,
396422
aarch64-datascience,
397423
aarch64-pyspark,
398424
aarch64-all-spark,

0 commit comments

Comments
 (0)