Skip to content

GitLab

Unverified Commit 320b81de authored by jberchtold-nvidia's avatar jberchtold-nvidia Committed by GitHub
Browse files

Update README to list JAX NGC container (#1756) 


Signed-off-by: default avatarJeremy Berchtold <jberchtold@nvidia.com>
parent 26ecb2f1
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 0 deletions
README.rst
View file @ 320b81de
...@@ -170,12 +170,20 @@ Docker (Recommended) ...@@ -170,12 +170,20 @@ Docker (Recommended)
^^^^^^^^^^^^^^^^^^^ ^^^^^^^^^^^^^^^^^^^
The quickest way to get started with Transformer Engine is by using Docker images on The quickest way to get started with Transformer Engine is by using Docker images on
`NVIDIA GPU Cloud (NGC) Catalog <https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch>`_. `NVIDIA GPU Cloud (NGC) Catalog <https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch>`_.
For example to use the NGC PyTorch container interactively, For example to use the NGC PyTorch container interactively,
.. code-block:: bash .. code-block:: bash
docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:25.04-py3 docker run --gpus all -it --rm nvcr.io/nvidia/pytorch:25.04-py3
For example to use the NGC JAX container interactively,
.. code-block:: bash
docker run --gpus all -it --rm nvcr.io/nvidia/jax:25.04-py3
Where 25.04 (corresponding to April 2025 release) is the container version. Where 25.04 (corresponding to April 2025 release) is the container version.
**Benefits of using NGC containers:** **Benefits of using NGC containers:**
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment