Software Component Versions

User Guide (Latest Version)

This section identifies the versions of software components and services that are used in this release of the NVIDIA NeMo™ Framework.

Software Component

Version

PyTorch 2.3.0a0
PyTorch Lightning 2.2.4
Megatron Core 0.7.0
Transformer Engine 1.6.0.dev0
NeMo 2.0.0rc0
NeMo Aligner 0.3.0.dev0
NeMo Data Curator 0.3.0

NVIDIA NeMo™ Framework Training container is built on top of NVIDIA Optimized Frameworks PyTorch 24.02 container: https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/index.html

Software Component

Version

Transformer Engine v1.3
PyTorch 2.2.0
NeMo 1.23.0
NeMo Aligner 0.2.0
NeMo Data Curator 0.1.0
Megatron Core 0.5.0
PyTorch Lightning 2.0.7
Hydra 1.2.0
Kubernetes 1.27.4
Helm 3.12.1
GPU Operator 23.3.2
Network Operator 23.1.0
KubeFlow Operator 1.6.0

NVIDIA NeMo™ Framework Training container is built on top of NVIDIA Optimized Frameworks PyTorch 24.01 container: https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/index.html

Software Component

Version

Transformer Engine v1.1
PyTorch 2.1.0
NeMo 1.22.0
Megatron Core 0.4.0
PyTorch Lightning 2.0.7
Hydra 1.2.0
Kubernetes 1.27.4
Helm 3.12.1
GPU Operator 23.3.2
Network Operator 23.1.0
KubeFlow Operator 1.6.0

NVIDIA NeMo™ Framework Training container is built on top of NVIDIA Optimized Frameworks PyTorch 23.11 container: https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/index.html

Previous Why NeMo Framework?
Next Getting Started
© | | | | | | |. Last updated on May 30, 2024.