forked from NVIDIA/gpu-operator
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathplugin-test.yaml
35 lines (35 loc) · 965 Bytes
/
plugin-test.yaml
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
27
28
29
30
31
32
33
34
35
apiVersion: apps/v1
kind: Deployment
metadata:
name: nvidia-plugin-test
labels:
app: nvidia-plugin-test
spec:
replicas: 5
selector:
matchLabels:
app: nvidia-plugin-test
template:
metadata:
labels:
app: nvidia-plugin-test
spec:
tolerations:
- key: nvidia.com/gpu
operator: Exists
effect: NoSchedule
serviceAccountName: nvidia-device-plugin
containers:
- name: nvidia-plugin-test-ctr
image: nvcr.io/nvidia/cloud-native/gpu-operator-validator:v1.10.1
imagePullPolicy: IfNotPresent
command: ['sh', '-c']
args:
- "while true; do vectorAdd; sleep 30; done"
securityContext:
allowPrivilegeEscalation: false
resources:
limits:
nvidia.com/gpu: 1
nodeSelector: # Schedule on the node with GPU sharing enabled
nvidia.com/gpu.product: Tesla-T4-SHARED