wandb

1
---
2
kind: PersistentVolume
3
apiVersion: v1
4
metadata:
5
  name: gpu-pod-results-volume
6
spec:
7
  storageClassName: gpu-pod-pv
8
  capacity:
9
    storage: 1Gi
10
  accessModes:
11
    - ReadWriteOnce
12
  hostPath:
13
    path: "/tmp"
14

15
---
16
kind: PersistentVolumeClaim
17
apiVersion: v1
18
metadata:
19
  name: gpu-pod-results-volumeclaim
20
spec:
21
  storageClassName: gpu-pod-pv
22
  accessModes:
23
    - ReadWriteOnce
24
  resources:
25
    requests:
26
      storage: 100Mi
27

28
---
29
kind: Pod
30
apiVersion: v1
31
metadata:
32
  name: gpu-pod
33
spec:
34
  restartPolicy: Never
35
  containers:
36
    - name: gpu-sdk
37
      image: gcr.io/wandb-client-cicd/gpu-sdk:latest
38
      env:
39
        - name: WANDB_API_KEY
40
          value: WANDB_API_KEY_PLACEHOLDER
41
      resources:
42
        limits:
43
          cpu: 3.999
44
          nvidia.com/gpu: 2
45
        requests:
46
          cpu: 3.5
47
      volumeMounts:
48
        - mountPath: "/wandb-store"
49
          name: gpu-pod-results-volumeclaim-name
50
  volumes:
51
    - name: gpu-pod-results-volumeclaim-name
52
      persistentVolumeClaim:
53
       claimName: gpu-pod-results-volumeclaim
54