Merlin Containers#
Merlin and the Merlin component libraries are available in Docker containers from the NVIDIA GPU Cloud (NCG) catalog. Access the catalog of containers at https://catalog.ngc.nvidia.com/containers.
The following table identifies the container names, catalog URL, and key Merlin components.
Container Name |
NGC Catalog URL |
Key Merlin Components |
|---|---|---|
merlin-hugectr |
https://catalog.ngc.nvidia.com/orgs/nvidia/teams/merlin/containers/merlin-hugectr |
Merlin libraries, including HugeCTR |
merlin-tensorflow |
https://catalog.ngc.nvidia.com/orgs/nvidia/teams/merlin/containers/merlin-tensorflow |
Merlin libraries, TensorFlow, and HugeCTR Tensorflow Embedding plugin |
merlin-pytorch |
https://catalog.ngc.nvidia.com/orgs/nvidia/teams/merlin/containers/merlin-pytorch |
Merlin libraries and PyTorch |
To use these containers, you must install the NVIDIA Container Toolkit to provide GPU support for Docker. You can use the NGC links referenced in the preceding table for more information about how to launch and run these containers.
Structural Changes Beginning with the 22.06 Releases#
With the 22.06 release of the Merlin containers, each container can perform model training as well as inference. Before the 22.06 release, the NGC catalog included one container for training and a separate container for inference.