Azure matrix 1gpu

perfkitbenchmarker/data/container/kubernetes_scale/kubernetes_scale.yaml.j2

@@ -53,3 +53,47 @@ spec:
         operator: "Exists"
         effect: "NoExecute"
         tolerationSeconds: {{ PodTimeout }}
+---
+{%- if NvidiaGpuRequest and cloud == "Azure" %}
+# According to the official Microsoft documentation, the NVIDIA device plugin
+# must be deployed as a DaemonSet to enable GPU support in the Kubernetes cluster.
+# Reference: https://learn.microsoft.com/en-us/azure/aks/use-nvidia-gpu?tabs=add-ubuntu-gpu-node-pool#nvidia-device-plugin-installation
+apiVersion: apps/v1
+kind: DaemonSet
+metadata:
+  name: nvidia-device-plugin-daemonset
+  namespace: kube-system
+spec:
+  selector:
+    matchLabels:
+      name: nvidia-device-plugin-ds
+  updateStrategy:
+    type: RollingUpdate
+  template:
+    metadata:
+      labels:
+        name: nvidia-device-plugin-ds
+    spec:
+      tolerations:
+      - key: 'nvidia.com/gpu'
+        operator: Exists
+        effect: NoSchedule
+      priorityClassName: 'system-node-critical'
+      containers:
+      - image: nvcr.io/nvidia/k8s-device-plugin:v0.18.0
+        name: nvidia-device-plugin-ctr
+        env:
+          - name: FAIL_ON_INIT_ERROR
+            value: 'false'
+        securityContext:
+          allowPrivilegeEscalation: false
+          capabilities:
+            drop: ['ALL']
+        volumeMounts:
+        - name: device-plugin
+          mountPath: /var/lib/kubelet/device-plugins
+      volumes:
+      - name: device-plugin
+        hostPath:
+          path: /var/lib/kubelet/device-plugins
+{%- endif %}

perfkitbenchmarker/linux_benchmarks/kubernetes_scale_benchmark.py

@@ -192,6 +192,7 @@ def ScaleUpPods(
       EphemeralStorageRequest='10Mi',
       RolloutTimeout=max_wait_time,
       PodTimeout=resource_timeout,
+      cloud=FLAGS.cloud,
   )
   cluster.ModifyPodSpecPlacementYaml(
       yaml_docs,