diff --git a/docs/source/examples.rst b/docs/source/examples.rst index 33d03f92..4c435bb2 100644 --- a/docs/source/examples.rst +++ b/docs/source/examples.rst @@ -4,9 +4,9 @@ Examples The examples below run on the `torchelastic/examples `_ Docker image, built from the `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 -------------- diff --git a/examples/Dockerfile b/examples/Dockerfile index 606a0c65..74ba7c81 100644 --- a/examples/Dockerfile +++ b/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 diff --git a/kubernetes/config/samples/classy-vision.yaml b/kubernetes/config/samples/classy-vision.yaml index 7a16e701..c043d710 100644 --- a/kubernetes/config/samples/classy-vision.yaml +++ b/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" diff --git a/kubernetes/config/samples/imagenet.yaml b/kubernetes/config/samples/imagenet.yaml index 0fec87e2..65c79141 100644 --- a/kubernetes/config/samples/imagenet.yaml +++ b/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"