Move arm64 wheel builds to GitHub runners

GitHub has experimental ARM64 runners.

Torch has also reached version 2.7.0.

Author: kohtala

.github/workflows/torch-wheel.yml

@@ -8,9 +8,9 @@ on:
   workflow_dispatch:
     inputs:
         pytorch-version:
-            description: 'Version of pytorch to build (e.g. 2.3.1)'
+            description: 'Version of pytorch to build (e.g. 2.7.0)'
             required: true
-            default: '2.3.1'
+            default: '2.7.0'
             type: string
         push-to-registry:
             description: 'Push image to registry'
@@ -25,7 +25,7 @@ permissions:
 
 jobs:
   build-pytorch-wheel:
-    runs-on: [self-hosted, ARM64]
+    runs-on: ubuntu-24.04-arm
     timeout-minutes: 1439  # GITHUB_TOKEN expires in 24 hours, so keep it just below that
     steps:
     - name: Print Inputs

.github/workflows/torchvision-wheel.yml

@@ -9,14 +9,14 @@ on:
   workflow_dispatch:
     inputs:
         torchvision-version:
-            description: 'Version of torchvision to build (e.g. 0.18.1)'
+            description: 'Version of torchvision to build (e.g. 0.22.0)'
             required: true
-            default: '0.18.1'
+            default: '0.22.0'
             type: string
         pytorch-version:
             description: 'Torch version (check compatibility matrix)'
             required: true
-            default: '2.3.1'
+            default: '2.7.0'
             type: string
         push-to-registry:
             description: 'Push image to registry'
@@ -31,7 +31,7 @@ permissions:
 
 jobs:
   build-torchvision-wheel:
-    runs-on: [self-hosted, ARM64]
+    runs-on: ubuntu-24.04-arm
     timeout-minutes: 1439  # GITHUB_TOKEN expires in 24 hours, so keep it just below that
     steps:
     - name: Print Inputs

torch-wheel.Dockerfile

@@ -1,4 +1,5 @@
-ARG PYTORCH_BUILD_VERSION=2.3.1
+ARG PYTORCH_BUILD_VERSION=2.7.0
+# https://catalog.ngc.nvidia.com/orgs/nvidia/containers/l4t-cuda/tags
 ARG BASE_IMAGE=nvcr.io/nvidia/l4t-cuda:12.2.12-devel
 
 # Builder stage for x86 wheel, just a dummy for now

torchvision-wheel.Dockerfile

@@ -1,7 +1,8 @@
 # Compatibility matrix for torch+torchvision: https://github.com/pytorch/vision
-ARG TORCHVISION_VERSION=0.18.1
-ARG PYTORCH_VERSION=2.3.1
+ARG TORCHVISION_VERSION=0.22.0
+ARG PYTORCH_VERSION=2.7.0
 ARG TORCH_WHEEL_SOURCE="scratch"
+# https://catalog.ngc.nvidia.com/orgs/nvidia/containers/l4t-cuda/tags
 ARG ARM_BASE_IMAGE=nvcr.io/nvidia/l4t-cuda:12.2.12-devel