-
Notifications
You must be signed in to change notification settings - Fork 10
Expand file tree
/
Copy pathnvidia-smi-mcdm.yml
More file actions
26 lines (25 loc) · 888 Bytes
/
nvidia-smi-mcdm.yml
File metadata and controls
26 lines (25 loc) · 888 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
# Example Job for running the NVIDIA SMI tool inside a container
#
# This version of the Job requests a compute-only device from the MCDM device plugin. For a version that
# requests a display device from the WDDM device plugin, see the file `nvidia-smi-wddm.yml`
#
# NOTE: this Job will only work when the device allocated by the MCDM device plugin is an NVIDIA GPU,
# otherwise the executable `nvidia-smi.exe` won't exist and the Pod will fail to start.
apiVersion: batch/v1
kind: Job
metadata:
name: example-nvidia-smi
spec:
template:
spec:
containers:
- name: example-nvidia-smi
image: "mcr.microsoft.com/windows/servercore:ltsc2022"
command: ["nvidia-smi.exe"]
resources:
limits:
directx.microsoft.com/compute: 1
nodeSelector:
"kubernetes.io/os": windows
restartPolicy: Never
backoffLimit: 0