重要提示

您正在查看 NeMo 2.0 文档。此版本引入了 API 的重大更改和一个新的库 NeMo Run。我们目前正在将 NeMo 1.0 的所有功能移植到 2.0。有关先前版本或 2.0 中尚不可用的功能的文档，请参阅 NeMo 24.07 文档。

软件组件版本#

本节介绍此版本 NVIDIA NeMo™ 框架中使用的软件组件和服务的版本。

NeMo 框架 24.12#

软件组件	版本
PyTorch	2.5.0a0
PyTorch Lightning	2.5.0.post0
Megatron Core	0.10.0
Transformer Engine	1.11
NeMo	2.1.0
NeMo Run	0.2.0
NeMo 对齐器	0.6.0
NeMo Data Curator	0.6.0
TRT-LLM	0.15.0

NeMo 框架 24.09#

软件组件	版本
PyTorch	2.4.0a0
PyTorch Lightning	2.4.0
Megatron Core	0.9.0
Transformer Engine	1.10
NeMo	2.0.0
NeMo 对齐器	0.5.0
NeMo Data Curator	0.5.0
TRT-LLM	0.12.0

NeMo 框架 24.07#

软件组件	版本
PyTorch	2.3.0a0
PyTorch Lightning	2.3.3
Megatron Core	0.8
Transformer Engine	1.8.0.dev0
NeMo	2.0.0rc1
NeMo 对齐器	0.4.0rc0
NeMo Data Curator	0.4.0
TRT-LLM	0.10.0

NeMo 框架 24.05#

软件组件	版本
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 对齐器	0.3.0.dev0
NeMo Data Curator	0.3.0

NVIDIA NeMo™ 框架训练容器构建于 NVIDIA 优化框架 PyTorch 24.02 容器之上：http://docs.nvda.net.cn/deeplearning/frameworks/pytorch-release-notes/index.html

NeMo 框架 24.01#

软件组件	版本
Transformer Engine	v1.3
PyTorch	2.2.0
NeMo	1.23.0
NeMo 对齐器	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™ 框架训练容器构建于 NVIDIA 优化框架 PyTorch 24.01 容器之上：http://docs.nvda.net.cn/deeplearning/frameworks/pytorch-release-notes/index.html

NeMo 框架 23.11#

软件组件	版本
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™ 框架训练容器构建于 NVIDIA 优化框架 PyTorch 23.11 容器之上：http://docs.nvda.net.cn/deeplearning/frameworks/pytorch-release-notes/index.html