forked from jina-ai/discoart
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpod.yaml
34 lines (34 loc) · 887 Bytes
/
pod.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
apiVersion: v1
kind: Pod
metadata:
name: disco
namespace: tenant-mobilecoin-imogen
spec:
containers:
- name: disco
image: technillogue/personality-clip:0.0.2
# command: ["/usr/bin/sleep"]
# args: ["1h"]
resources:
limits:
cpu: 1
memory: 4Gi #idk if it needs this much
nvidia.com/gpu: 1
affinity:
nodeAffinity:
requiredDuringSchedulingIgnoredDuringExecution:
nodeSelectorTerms:
- matchExpressions:
- key: gpu.nvidia.com/class
operator: In
values:
- Tesla_V100_PCIE
- Tesla_V100_NVLINK
preferredDuringSchedulingIgnoredDuringExecution:
- weight: 20
preference:
matchExpressions:
- key: gpu.nvidia.com/class
operator: In
values:
- Tesla_V100_PCIE