Skip to content

Commit 3820ce9

Browse files
committed
✅ Update tutorial job
1 parent e8771be commit 3820ce9

1 file changed

Lines changed: 32 additions & 0 deletions

File tree

mnist/tutorial-job.yaml

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
apiVersion: batch/v1
2+
kind: Job
3+
metadata:
4+
name: tutorial-job
5+
spec:
6+
template:
7+
spec:
8+
containers:
9+
- name: gpu-container
10+
image: gitlab-registry.nrp-nautilus.io/zihaozhou/nautilus_tutorial:jupyterhub
11+
command: ["/bin/bash","-c"]
12+
args: ["mkdir MNIST; cd MNIST;
13+
wget https://raw.githubusercontent.com/pytorch/examples/main/mnist/main.py;
14+
python main.py --save-model;
15+
mv mnist_cnn.pt /tutorial"]
16+
volumeMounts:
17+
- mountPath: /tutorial
18+
name: tutorial-volume
19+
resources:
20+
limits:
21+
nvidia.com/gpu: "1"
22+
memory: "8G"
23+
cpu: "4"
24+
requests:
25+
nvidia.com/gpu: "1"
26+
memory: "8G"
27+
cpu: "4"
28+
restartPolicy: Never
29+
volumes:
30+
- name: tutorial-volume
31+
persistentVolumeClaim:
32+
claimName: tutorial-volume

0 commit comments

Comments
 (0)