Skip to content

Instantly share code, notes, and snippets.

@vpavlin
Created June 2, 2020 19:17
Show Gist options
  • Save vpavlin/ecbfeb23bb934c18ee9361e60ef1f9b7 to your computer and use it in GitHub Desktop.
Save vpavlin/ecbfeb23bb934c18ee9361e60ef1f9b7 to your computer and use it in GitHub Desktop.
apiVersion: "kubeflow.org/v1"
kind: "PyTorchJob"
metadata:
name: "pytorch-dist-mnist-gloo"
spec:
pytorchReplicaSpecs:
Master:
replicas: 1
restartPolicy: OnFailure
template:
spec:
containers:
- name: pytorch
image: quay.io/vpavlin/pytorch-dist-mnist-test:1.0-3
args: ["--backend", "gloo"]
resources:
limits:
memory: 1Gi
cpu: 1
Worker:
replicas: 1
restartPolicy: OnFailure
template:
spec:
containers:
- name: pytorch
image: quay.io/vpavlin/pytorch-dist-mnist-test:1.0-3
args: ["--backend", "gloo"]
resources:
limits:
memory: 1Gi
cpu: 1
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment