This chapter identifies the versions of software components and services that are used in this release of the NVIDIA NeMo™ framework.
Software Component |
Version |
|---|---|
| Transformer Engine | v0.12 |
| PyTorch | 2.1.0 |
| NeMo | 1.21.0 |
| Megatron Core | 0.3.0 |
| PyTorch Lightning | 2.0.7 |
| Hydra | 1.3.0 |
| Kubernetes | 1.27.4 |
| Helm | 3.12.1 |
| GPU Operator | 23.3.2 |
| Network Operator | 23.1.0 |
| KubeFlow Operator | 1.6.0 |
| TensorRT | 9.1.0.4 |
| TensorRT-LLM | 0.5.0 |
| PyTriton | 0.3.1 |
| Triton | 23.08 |
| NeMo | 1.21.0 |
NVIDIA NeMo™ framework container is built on top of PyTorch 23.08 container: https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/index.html