25 lines
507 B
YAML
25 lines
507 B
YAML
|
---
|
||
|
apiVersion: apps/v1
|
||
|
kind: DaemonSet
|
||
|
metadata:
|
||
|
namespace: default
|
||
|
name: gpu-pod
|
||
|
spec:
|
||
|
selector:
|
||
|
matchLabels:
|
||
|
app: gpu-pod
|
||
|
template:
|
||
|
metadata:
|
||
|
labels:
|
||
|
app: gpu-pod
|
||
|
spec:
|
||
|
nodeSelector:
|
||
|
nvidia.com/gpu.family: pascal
|
||
|
containers:
|
||
|
- name: gpu-pod
|
||
|
image: nvidia/cuda:12.3.1-runtime-ubuntu20.04
|
||
|
command: ["sleep", "infinity"]
|
||
|
terminationGracePeriodSeconds: 0
|
||
|
runtimeClassName: nvidia
|
||
|
restartPolicy: Always
|