NVIDIA
NVIDIA
NeMo Framework Megatron Backend
Container
NVIDIA
NVIDIA
NeMo Framework Megatron Backend

NVIDIA NeMo™ framework Megatron backend supports pre-training, post-training, and reinforcement learning of LLMs and multi-modal generative AI models with state-of-the-art data processing, model training techniques, and flexible deployment options.

LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.ref=
02/26/2025 10:55 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_REF
02/26/2025 10:55 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.id=134983854
02/26/2025 10:55 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NVIDIA_BUILD_ID=134983854
02/26/2025 10:55 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_ID
02/26/2025 10:55 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NEMO_FW_PIPELINE_ID=24597396
02/26/2025 10:55 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NEMO_FW_PIPELINE_ID=24597396
02/26/2025 10:55 AM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1RUN
NVIDIA_BIGNLP_VERSION= TRT_VER=10.8.0.24 CUDA_VER=12.8 CUDNN_VER=9.5.1.17-1 NCCL_VER= CUBLAS_VER=12.6.1.4-1 NVRTC_VER=12.6.85-1 TRT_LLM_COMMIT=2830e4e8099c0d7e6402494865b9cb7433f51eb5 TRT_LLM_VERSION=0.17.0 REINSTALL_APEX=False APEX_COMMIT=b216eeee7dd91478745496919e9c0167cc4c41e3 MCORE_COMMIT=791b70e031dc6905f77e29c25f1e2ac5b8dd4870 MCORE_VERSION=0.11.0rc1 SRC_DIR=/src/tensorrt_llm CPP_BUILD_DIR=/src/tensorrt_llm/cpp/build TARGET_ARCH=arm INSTALL_VLLM=False ALIGNER_COMMIT=r0.7.0 ALIGNER_VERSION=0.7.0rc1.dev0 LAUNCHER_COMMIT=4abd481402adc8a061942c486eda6d71f19de718 DATA_CURATOR_COMMIT=r0.7.0 INSTALL_DATA_CURATOR=True DATA_CURATOR_VERSION=0.7.0rc2.dev0 INSTALL_MODELOPT=False MODELOPT_VERSION= NEMO_COMMIT=r2.2.0 NEMO_VERSION=2.2.0rc4 INSTALL_RESIL=True RESIL_COMMIT=b6eb61dbf9fe272b1a943b1b0d9efdde99df0737 RESIL_VERSION=0.2.0 NEMO_RUN_COMMIT=451f781c67698582b7ba91ac86aac0e5b5c6ae52 PRE_COMPILE_QUANT_KERNELS=False chmod -R a+w /workspace
02/26/2025 10:55 AM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
/workspace
02/26/2025 10:55 AM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1RUN
NVIDIA_BIGNLP_VERSION= TRT_VER=10.8.0.24 CUDA_VER=12.8 CUDNN_VER=9.5.1.17-1 NCCL_VER= CUBLAS_VER=12.6.1.4-1 NVRTC_VER=12.6.85-1 TRT_LLM_COMMIT=2830e4e8099c0d7e6402494865b9cb7433f51eb5 TRT_LLM_VERSION=0.17.0 REINSTALL_APEX=False APEX_COMMIT=b216eeee7dd91478745496919e9c0167cc4c41e3 MCORE_COMMIT=791b70e031dc6905f77e29c25f1e2ac5b8dd4870 MCORE_VERSION=0.11.0rc1 SRC_DIR=/src/tensorrt_llm CPP_BUILD_DIR=/src/tensorrt_llm/cpp/build TARGET_ARCH=arm INSTALL_VLLM=False ALIGNER_COMMIT=r0.7.0 ALIGNER_VERSION=0.7.0rc1.dev0 LAUNCHER_COMMIT=4abd481402adc8a061942c486eda6d71f19de718 DATA_CURATOR_COMMIT=r0.7.0 INSTALL_DATA_CURATOR=True DATA_CURATOR_VERSION=0.7.0rc2.dev0 INSTALL_MODELOPT=False MODELOPT_VERSION= NEMO_COMMIT=r2.2.0 NEMO_VERSION=2.2.0rc4 INSTALL_RESIL=True RESIL_COMMIT=b6eb61dbf9fe272b1a943b1b0d9efdde99df0737 RESIL_VERSION=0.2.0 NEMO_RUN_COMMIT=451f781c67698582b7ba91ac86aac0e5b5c6ae52 PRE_COMPILE_QUANT_KERNELS=False if [ $PRE_COMPILE_QUANT_KERNELS = "True" ]; then python -c "import modelopt.torch.quantization.extensions as ext; ext.precompile()"; fi
02/26/2025 10:55 AM UTC
...