Skip to content

Commit 1938101

Browse files
committed
build for aarch64-pytorch-cuda11 as well (for GH100 and simplified CI config)
1 parent 3817d82 commit 1938101

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

.github/workflows/docker.yml

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -260,6 +260,18 @@ jobs:
260260
needs: x86_64-scipy
261261
if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }}
262262

263+
aarch64-pytorch-cuda11:
264+
uses: ./.github/workflows/docker-build-test-upload.yml
265+
with:
266+
parent-image: scipy-notebook
267+
image: pytorch-notebook
268+
variant: cuda11
269+
platform: aarch64
270+
runs-on: ubuntu-24.04-arm
271+
timeout-minutes: 25
272+
needs: aarch64-scipy
273+
if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }}
274+
263275
x86_64-pytorch-cuda11:
264276
uses: ./.github/workflows/docker-build-test-upload.yml
265277
with:
@@ -418,6 +430,7 @@ jobs:
418430
aarch64-tensorflow,
419431
aarch64-tensorflow-cuda,
420432
aarch64-pytorch,
433+
aarch64-pytorch-cuda11,
421434
aarch64-pytorch-cuda12,
422435
aarch64-datascience,
423436
aarch64-pyspark,

0 commit comments

Comments
 (0)