From b6b4a9d9fa10573eb480595dd0bf2691e8e00664 Mon Sep 17 00:00:00 2001
From: Dick Tang <dick.tang@ymail.com>
Date: Tue, 18 Nov 2025 15:04:53 +0800
Subject: [PATCH 1/2] add aarch64 cuda12

---
 .github/workflows/docker.yml | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index ed5f407cfc..f5b7ce5dd7 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
+      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,

From 93a7d1d68a94c088cddb62ba926c5a578bbbb265 Mon Sep 17 00:00:00 2001
From: Dick Tang <dick.tang@ymail.com>
Date: Tue, 18 Nov 2025 17:53:56 +0800
Subject: [PATCH 2/2] Change runner to ubuntu-24.04-arm for aarch64

---
 .github/workflows/docker.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index f5b7ce5dd7..3aeae996a7 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -267,7 +267,7 @@ jobs:
       image: pytorch-notebook
       variant: cuda12
       platform: aarch64
-      runs-on: ubuntu-24.04
+      runs-on: ubuntu-24.04-arm
       timeout-minutes: 25
     needs: aarch64-scipy
     if: ${{ !contains(github.event.pull_request.title, '[FAST_BUILD]') }}