Skip to content

Commit

Permalink
verify and add docker building instruction (NVIDIA#3236)
Browse files Browse the repository at this point in the history
Signed-off-by: fayejf <[email protected]>
  • Loading branch information
fayejf authored Nov 23, 2021
1 parent 0178ff7 commit 05316d4
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 0 deletions.
5 changes: 5 additions & 0 deletions README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -178,6 +178,11 @@ Megatron GPT training requires NVIDIA Apex to be installed.
Docker containers:
~~~~~~~~~~~~~~~~~~
To build a nemo container with Dockerfile from a branch, please run

.. code-block:: bash
DOCKER_BUILDKIT=1 docker build -f Dockerfile -t nemo:latest .
If you chose to work with main branch, we recommend using NVIDIA's PyTorch container version 21.11-py3 and then installing from GitHub.

Expand Down
6 changes: 6 additions & 0 deletions docs/source/starthere/intro.rst
Original file line number Diff line number Diff line change
Expand Up @@ -134,6 +134,12 @@ Use this installation mode if you are contributing to NeMo.
Docker containers
~~~~~~~~~~~~~~~~~
To build a nemo container with Dockerfile from a branch, please run

.. code-block:: bash
DOCKER_BUILDKIT=1 docker build -f Dockerfile -t nemo:latest.
If you chose to work with the ``main`` branch, we recommend using `NVIDIA's PyTorch container version 21.05-py3 <https://ngc.nvidia.com/containers/nvidia:pytorch/tags>`_, then install from GitHub.

.. code-block:: bash
Expand Down

0 comments on commit 05316d4

Please sign in to comment.