apiVersion: v1
kind: Pod
metadata:
  name: kube-rn50-1
spec:
  containers:
  - name: tf-mnist-1gpu
    image: centaurusinfra/tf-onegpu-mnist:latest
    workingDir: /tmp/test
    command: ["sh", "-c", "python3 tf-cf-rn50-pack.py"]
  restartPolicy: OnFailure
