Support building container image on aarch64 (arm64)

Signed-off-by: Dr. Jan-Philip Gehrcke <[email protected]>

Author: jgehrcke

deployments/container/Dockerfile

@@ -14,16 +14,19 @@
 
 ARG GOLANG_VERSION=1.23.8
 # We use an ubuntu20.04 base image to allow for a more efficient multi-arch builds.
-FROM --platform=${BUILDOS}/amd64 nvcr.io/nvidia/cuda:12.8.1-base-ubuntu20.04 AS build
+FROM  nvcr.io/nvidia/cuda:12.8.1-base-ubuntu20.04 AS build
 
 RUN apt-get update && \
     apt-get install -y wget make git gcc-aarch64-linux-gnu gcc \
     && \
     rm -rf /var/lib/apt/lists/*
 
 ARG GOLANG_VERSION=x.x.x
-RUN wget -nv -O - https://storage.googleapis.com/golang/go${GOLANG_VERSION}.linux-amd64.tar.gz \
-    | tar -C /usr/local -xz
+
+# Support building on Linux on aarch64 (arm64) and on x86_64 (amd64).
+RUN _ARCH="arm64"; if [ "$(uname -m)" = "x86_64" ]; then _ARCH="amd64"; fi; \
+    wget -nv -O - https://storage.googleapis.com/golang/go${GOLANG_VERSION}.linux-${_ARCH}.tar.gz \
+        | tar -C /usr/local -xz
 
 ENV GOPATH /go
 ENV PATH $GOPATH/bin:/usr/local/go/bin:$PATH
@@ -59,8 +62,8 @@ LABEL com.nvidia.git-commit="${GIT_COMMIT}"
 LABEL release="N/A"
 LABEL summary="NVIDIA DRA Driver for GPUs"
 LABEL description="NVIDIA DRA Driver for GPUs"
-LABEL org.opencontainers.image.description "NVIDIA DRA Driver for GPUs"
-LABEL org.opencontainers.image.source "https://github.com/NVIDIA/k8s-dra-driver-gpu"
+LABEL org.opencontainers.image.description="NVIDIA DRA Driver for GPUs"
+LABEL org.opencontainers.image.source="https://github.com/NVIDIA/k8s-dra-driver-gpu"
 
 RUN mkdir /licenses && mv /NGC-DL-CONTAINER-LICENSE /licenses/NGC-DL-CONTAINER-LICENSE