Bump version to 0.2.0 in docker container

Summary: Bump version to 0.2.0 in docker container

Reviewed By: kiukchung

Differential Revision: D21316175

fbshipit-source-id: ac0471164e2d5817d5529d629d50a30cb25966d7

Author: aivanou

docs/source/examples.rst

@@ -4,9 +4,9 @@ Examples
 The examples below run on the `torchelastic/examples <https://hub.docker.com/r/torchelastic/examples>`_
 Docker image, built from the `examples/Dockerfile <https://github.com/pytorch/elastic/blob/master/examples/Dockerfile>`_.
 
-.. note:: The ``$VERSION`` (e.g. ``0.2.0rc1``) variable is used throughout this page,
+.. note:: The ``$VERSION`` (e.g. ``0.2.0``) variable is used throughout this page,
           this should be substituted with the version of torchelastic you are using.
-          The examples below only work on torchelastic ``>0.2.0rc1``.
+          The examples below only work on torchelastic ``>=0.2.0``.
 
 Prerequisite
 --------------

examples/Dockerfile

@@ -1,4 +1,4 @@
-ARG BASE_IMAGE=pytorch/elastic:0.2.0rc1
+ARG BASE_IMAGE=pytorch/elastic:0.2.0
 FROM $BASE_IMAGE
 
 # install utilities and dependencies

kubernetes/config/samples/classy-vision.yaml

@@ -18,7 +18,7 @@ spec:
         spec:
           containers:
             - name: elasticjob-worker
-              image: torchelastic/examples:0.2.0rc1
+              image: torchelastic/examples:0.2.0
               imagePullPolicy: Always
               args:
                 - "--nproc_per_node=1"

kubernetes/config/samples/imagenet.yaml

@@ -18,7 +18,7 @@ spec:
         spec:
           containers:
             - name: elasticjob-worker
-              image: torchelastic/examples:0.2.0rc1
+              image: torchelastic/examples:0.2.0
               imagePullPolicy: Always
               args:
                 - "--nproc_per_node=1"