diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml index ed5f407cfc..3aeae996a7 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-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-cuda11: 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,