Support Data Center precompiled driver container for Arm (Ubuntu 24.04)

[shivakunv]

Code Changes Summary:

• Platform Support
  Added support for the ARM64 platform.
  AMD64 remains the default architecture.

• Artifacts Update
  ARM64 build artifacts are now uploaded with the -arm64 suffix.

• Instance Type and Region Mapping
  g4dn.xlarge:
  Architecture: AMD64
  Supported Region: us-west-1
  Used for AMD64 builds.
  g5g.xlarge:
  Architecture: ARM64
  Supported Region: us-west-2
  Used for ARM64 builds.

passed pipeline: https://github.com/NVIDIA/gpu-driver-container/actions/runs/22180871853

passed pipeline: https://github.com/NVIDIA/gpu-driver-container/actions/runs/22337833186

[Copilot]

Pull request overview

This pull request adds ARM64 (aarch64) platform support to the Ubuntu 24.04 precompiled driver container builds, while maintaining AMD64 as the default architecture. The changes enable multi-platform Docker builds and update the CI/CD pipeline to handle both architectures.

Changes:

• Added ARM64 platform support for Ubuntu 24.04 precompiled driver containers with architecture-specific package handling
• Updated CI workflow to build, test, and publish both AMD64 and ARM64 artifacts with platform-specific suffixes
• Modified Holodeck test infrastructure to support ARM64 instances (g5g.xlarge in us-west-2) and Ubuntu 24.04 OS specification

Reviewed changes

Copilot reviewed 10 out of 10 changed files in this pull request and generated 2 comments.

Show a summary per file

File	Description
ubuntu24.04/precompiled/nvidia-driver	Added conditional installation of libnvidia-fbc1 package (AMD64 only)
ubuntu24.04/precompiled/local-repo.sh	Added conditional downloads for ARM64-incompatible packages (linux-signatures-nvidia, libnvidia-fbc1)
ubuntu24.04/precompiled/Dockerfile	Made i386 architecture and CUDA repository URLs conditional based on target architecture
tests/scripts/findkernelversion.sh	Added optional PLATFORM_SUFFIX parameter for artifact matching and platform-specific manifest inspection
tests/scripts/ci-precompiled-helpers.sh	Added PLATFORM_SUFFIX parameter support for kernel version testing
tests/holodeck_ubuntu24.04.yaml	Removed file (merged into holodeck_ubuntu.yaml)
tests/holodeck_ubuntu.yaml	Removed hardcoded ingressIpRanges and AMI, added OS specification support
multi-arch.mk	Removed AMD64-only platform restriction for ubuntu24.04 builds
Makefile	Added DOCKER_BUILD_PLATFORM_OPTIONS to base image build targets
.github/workflows/precompiled.yaml	Added platform matrix dimension, platform-aware artifact naming, ARM64 e2e testing with appropriate instance types, and Holodeck version update

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

[tariq1890]

Why do we need skopeo?

[shivakunv]

Line 478 (Line: 356): Pushing the multi-arch oci-archive to the registry as a manifest list.
The github actions runner's docker load --platform exists in the cli but the daemon can not hold multi-arch images.
docker load + docker push would mean loop per platform, push each, then docker manifest create to stitch the manifest list.
skopeo copy streams oci-archive: registry preserving multi-arch in one step.

Line 379: Build outputs one multi-arch oci-archive (amd64 + arm64). The e2e test needs a single platform as docker-archive (for docker load).
On the github actions runner's docker save --platform does not help either, so without skopeo we would have to build amd64 and arm64 as separate single arch images. separate artifacts need to upload on github.
skopeo copy --override-arch extracts the platform we need.

consistency: same tool in both places.

From what I recall, tried regctl earlier and hit an oci-archive error. github pipeline logs have been cleared since three months have passed. can reinvestigate if preferred.

[tariq1890]

we would have to build amd64 and arm64 as separate single arch images. separate artifacts need to upload on github.

This is an approach we already follow in other repositories so I prefer this for the time being. If we want to use skopeo let's evaluate it first and then do it in a separate commit if we decide to move forward with skopeo

[shivakunv]

Thanks for the feedback.

we already agreed to move away from building separate per platform images and stitching them together.

please check:
#533 (comment): On putting more thought into this, can we do a multiarch build of the precompiled image instead of building the arm64 and amd64 images separate and then stitching them together?
#533 (comment): As mentioned in my earlier comment, let's do a single multiarch build instead of building individual platform-specific images and then merging them together.

for other repo or non-precompiled driver container:
Single buildx run pushes directly to registry with --output=type=image,push=true
no artifacts upload/download for e2e.

for precompiled repo (before change) :

-> build single arch image ( docker save image ) -> upload image as artifacts -> e2e: download version specific image artifact tar -> e2e: rsync tar image to target -> e2e: run gpu-operator on target with this tar image -> on Success publish image ( docker load + docker push platform specific amd64)

After change:

-> build multi arch image -> upload multi arch image as artifacts -> e2e: download version specific multi arch image artifact tar -> e2e: extract target specific (platform specific amd64 or arm64 ) image with skopeo -> e2e: rsync tar image to target -> e2e: run gpu-operatoron target with this tar image -> on Success publish multi arch image with skopeo

skopeo evaluation status:
skopeo was evaluated against the nv_only docker registry. Also, e2e tests are passing.

[rajathagasthya]

Let's get another approval and resolve all open threads before merging.

[tariq1890]

You can reduce the duplication here by conditionally installing only linux-objects-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION}

[shivakunv]

can you clarify ?
what I understood : ( only for linux-objects-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION})

if [ "$KERNEL_TYPE" != "kernel-open" ]; then
    apt-get install --no-install-recommends -y \
        linux-objects-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION}
fi
if [ "$KERNEL_TYPE" = "kernel-open" ]; then
    apt-get install --no-install-recommends -y \
        linux-modules-nvidia-${DRIVER_BRANCH}-server-open-${KERNEL_VERSION}
else
    apt-get install --no-install-recommends -y \
        linux-modules-nvidia-${DRIVER_BRANCH}-server-${KERNEL_VERSION}
fi

is more complex .