update to 21.04-py3

Signed-off-by: Oleksii Kuchaiev <okuchaiev@nvidia.com>

Dockerfile

@@ -14,7 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:21.03-py3
+ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:21.04-py3
 
 
 # build an image that includes only the nemo dependencies, ensures that dependencies
@@ -74,7 +74,7 @@ COPY . .
 
 # start building the final container
 FROM nemo-deps as nemo
-ARG NEMO_VERSION=1.0.0rc1
+ARG NEMO_VERSION=1.0.0
 
 # Check that NEMO_VERSION is set. Build will fail without this. Expose NEMO and base container
 # version information as runtime environment variable for introspection purposes

Jenkinsfile

@@ -1,7 +1,7 @@
 pipeline {
   agent {
         docker {
-      image 'nvcr.io/nvidia/pytorch:21.03-py3'
+      image 'nvcr.io/nvidia/pytorch:21.04-py3'
       args '--device=/dev/nvidia0 --gpus all --user 0:128 -v /home/TestData:/home/TestData -v $HOME/.cache/torch:/root/.cache/torch --shm-size=8g'
         }
   }

README.rst

@@ -50,7 +50,7 @@ Requirements
 ------------
 
 1) Python 3.6, 3.7 or 3.8
-2) Pytorch 1.8.0 or above
+2) Pytorch 1.8.1 or above
 3) NVIDIA GPU for training
 
 Documentation
@@ -137,13 +137,13 @@ It has all requirements and NeMo 1.0.0rc1 already installed.
     stack=67108864 --device=/dev/snd nvcr.io/nvidia/nemo:1.0.0rc1
 
 
-If you chose to work with main branch, we recommend using NVIDIA's PyTorch container version 21.03-py3 and then installing from GitHub.
+If you chose to work with main branch, we recommend using NVIDIA's PyTorch container version 21.04-py3 and then installing from GitHub.
 
 .. code-block:: bash
 
     docker run --gpus all -it --rm -v <nemo_github_folder>:/NeMo --shm-size=8g \
     -p 8888:8888 -p 6006:6006 --ulimit memlock=-1 --ulimit \
-    stack=67108864 --device=/dev/snd nvcr.io/nvidia/pytorch:21.03-py3
+    stack=67108864 --device=/dev/snd nvcr.io/nvidia/pytorch:21.04-py3
 
 Examples
 --------

docs/source/starthere/intro.rst

@@ -34,7 +34,7 @@ Before you begin using NeMo, it's assumed you meet the following prerequisites.
 
 #. You have Python version 3.6, 3.7 or 3.8.
 
-#. You have Pytorch version 1.7.1.
+#. You have Pytorch version 1.8.1 or above
 
 #. You have access to a Volta, Turing, or an NVIDIA Ampere architecture-based A100 GPU for training.
 
@@ -142,13 +142,13 @@ The container includes all the dependencies and NeMo 1.0.0b3 already installed.
     stack=67108864 --device=/dev/snd nvcr.io/nvidia/nemo:1.0.0rc1
 
 
-If you chose to work with the ``main`` branch, we recommend using `NVIDIA's PyTorch container version 20.11-py3 <https://ngc.nvidia.com/containers/nvidia:pytorch/tags>`_, then install from GitHub.
+If you chose to work with the ``main`` branch, we recommend using `NVIDIA's PyTorch container version 21.04-py3 <https://ngc.nvidia.com/containers/nvidia:pytorch/tags>`_, then install from GitHub.
 
 .. code-block:: bash
 
     docker run --gpus all -it --rm -v <nemo_github_folder>:/NeMo --shm-size=8g \
     -p 8888:8888 -p 6006:6006 --ulimit memlock=-1 --ulimit \
-    stack=67108864 --device=/dev/snd nvcr.io/nvidia/pytorch:20.11-py3
+    stack=67108864 --device=/dev/snd nvcr.io/nvidia/pytorch:21.04-py3
 
 
 FAQ