diff --git a/.github/workflows/docker-tag-push.yml b/.github/workflows/docker-tag-push.yml index b5e792016b..ecbce37d0c 100644 --- a/.github/workflows/docker-tag-push.yml +++ b/.github/workflows/docker-tag-push.yml @@ -48,7 +48,7 @@ jobs: image: ${{ inputs.image }} variant: ${{ inputs.variant }} platform: aarch64 - if: ${{ !contains(inputs.variant, 'cuda') }} + if: ${{ !contains(inputs.variant, 'cuda11') }} - 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..f9c50a7321 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: @@ -260,6 +272,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 +416,9 @@ jobs: aarch64-r, aarch64-julia, aarch64-tensorflow, + aarch64-tensorflow-cuda, aarch64-pytorch, + aarch64-pytorch-cuda12, aarch64-datascience, aarch64-pyspark, aarch64-all-spark,