File size: 727 Bytes
e78c13e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
36
37
38
39
apiVersion: v1
kind: Pod
metadata:
  name: photo-back2life
spec:
  containers:
    - name: photos-back2life
      image: <YOUR IMAGE>
      volumeMounts:
      - mountPath: /in
        name: in-folder
      - mountPath: /out
        name: out-folder
      command: 
        - python
        - /app/run.py
      args:
        - --input_folder
        - /in
        - --output_folder
        - /out
        - --GPU 
        - '0'
        - --with_scratch
      resources:
        limits:
          memory: 4Gi
          cpu: 0
          nvidia.com/gpu: 1      
  volumes:
  - name: in-folder
    hostPath:
      path: /srv/in
      type: Directory   
  - name: out-folder
    hostPath:
      path: /srv/out
      type: Directory