From ab41d411ce2864b4d2515c6917f3916ed108ca81 Mon Sep 17 00:00:00 2001 From: "roy.xue" Date: Mon, 23 Jul 2018 23:10:23 +0800 Subject: [PATCH] Add install pytorch operator step --- content/docs/started/pytorch.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/content/docs/started/pytorch.md b/content/docs/started/pytorch.md index b8460e3c4c..38d51fee85 100644 --- a/content/docs/started/pytorch.md +++ b/content/docs/started/pytorch.md @@ -37,6 +37,7 @@ If it is not included you can add it as follows ``` cd ${KSONNET_APP} +ks pkg install kubeflow/pytorch-job ks generate pytorch-operator pytorch-operator ks apply ${ENVIRONMENT} -c pytorch-operator ``` @@ -58,7 +59,7 @@ You should now be able to see the created pods matching the specified number of ``` kubectl get pods -l pytorch_job_name=dist-mnist-for-e2e-test ``` -Training should run for about 10 epochs and takes 5-10 minutes on a cpu cluster. Logs can be inspected to see its training progress. +Training should run for about 10 epochs and takes 5-10 minutes on a cpu cluster. Logs can be inspected to see its training progress. ``` PODNAME=$(kubectl get pods -l pytorch_job_name=dist-mnist-for-e2e-test,task_index=0 -o name)