diff --git a/.github/workflows/docker-tag-push.yml b/.github/workflows/docker-tag-push.yml index b0775eaf3b..ba02fedf02 100644 --- a/.github/workflows/docker-tag-push.yml +++ b/.github/workflows/docker-tag-push.yml @@ -48,7 +48,6 @@ jobs: image: ${{ inputs.image }} variant: ${{ inputs.variant }} platform: aarch64 - if: ${{ !contains(inputs.variant, 'cuda') }} - name: Download x86_64 image tar and apply tags 🏷 uses: ./.github/actions/apply-single-tags diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml index ed5f407cfc..fee47d52da 100644 --- a/.github/workflows/docker.yml +++ b/.github/workflows/docker.yml @@ -214,6 +214,18 @@ jobs: needs: x86_64-scipy if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }} + aarch64-tensorflow-cuda: + uses: ./.github/workflows/docker-build-test-upload.yml + with: + parent-image: scipy-notebook + image: tensorflow-notebook + variant: cuda + platform: aarch64 + runs-on: ubuntu-24.04-arm + timeout-minutes: 25 + needs: aarch64-scipy + if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }} + x86_64-tensorflow-cuda: uses: ./.github/workflows/docker-build-test-upload.yml with: @@ -248,6 +260,18 @@ jobs: needs: x86_64-scipy if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }} + aarch64-pytorch-cuda11: + uses: ./.github/workflows/docker-build-test-upload.yml + with: + parent-image: scipy-notebook + image: pytorch-notebook + variant: cuda11 + platform: aarch64 + runs-on: ubuntu-24.04-arm + timeout-minutes: 25 + needs: aarch64-scipy + if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }} + x86_64-pytorch-cuda11: uses: ./.github/workflows/docker-build-test-upload.yml with: @@ -260,6 +284,18 @@ jobs: needs: x86_64-scipy if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }} + aarch64-pytorch-cuda12: + uses: ./.github/workflows/docker-build-test-upload.yml + with: + parent-image: scipy-notebook + image: pytorch-notebook + variant: cuda12 + platform: aarch64 + runs-on: ubuntu-24.04-arm + timeout-minutes: 25 + needs: aarch64-scipy + if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }} + x86_64-pytorch-cuda12: uses: ./.github/workflows/docker-build-test-upload.yml with: @@ -392,7 +428,10 @@ jobs: aarch64-r, aarch64-julia, aarch64-tensorflow, + aarch64-tensorflow-cuda, aarch64-pytorch, + aarch64-pytorch-cuda11, + aarch64-pytorch-cuda12, aarch64-datascience, aarch64-pyspark, aarch64-all-spark,