ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
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.

NVIDIA AI Enterprise Supported
LayerLabelCreated
dcd26f4d1da12e0009929d5f7b3caddd45fe963ec53e23e3da2a8e997a92df8cCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
02/08/2025 12:08 AM UTC
c5831c0040ac3f274c13ccdc6560ba14181c8a2d983374d486d321741e931451RUN
git clone --single-branch --branch cosmos https://github.com/NVIDIA/Megatron-LM.git /opt/megatron-lm &&
  cd /opt/megatron-lm &&
  pip install -e . &&
  cd megatron/core/datasets &&
  make
02/08/2025 12:08 AM UTC
a3409f9b0e99c0a380bd93206254efe9beb657b99432e0dc2d4cd4a7e9ffe57dRUN
rm -rf /opt/megatron-lm
02/08/2025 12:07 AM UTC
87311597f73e43ad4b1f103ff2b7c5d39e4a8436a0ca8ffcb2ef3f7635696a5bRUN
pip install --no-cache-dir imageio[ffmpeg] pyav iopath better_profanity peft git+https://github.com/NVlabs/Pytorch_Retinaface.git@b843f45
01/05/2025 12:06 AM UTC
1ecd7dda86c2e375dbcef691233c0ca778e713fd26d5154f2214f9ad94f65cc7LABEL
com.nvidia.build.ref=
12/21/2024 7:06 PM UTC
c17bbd4d18c0117dd6ede0b89a9b5fade26b0702967fc3907c52d4af98a9f65dARG
NVIDIA_BUILD_REF
12/21/2024 7:06 PM UTC
6c1c3646b4b4b6caa05825e3bf849f82fa45097358e4fd0ecdc673cf0a348c69LABEL
com.nvidia.build.id=114410972
12/21/2024 7:06 PM UTC
7ab040bf46ae8296f38b86b510e461094fc404c13ca6337fbfa52ff2d74943feENV
NVIDIA_BUILD_ID=114410972
12/21/2024 7:06 PM UTC
326feeeae7b9674a432a7247ae372c7012ae0592ce53c4d189d76de0a2bbeefbARG
NVIDIA_BUILD_ID
12/21/2024 7:06 PM UTC
2044e72fb1094c328188f766ab0ac38f099f5da6c6b7f89342b7d23d212054fbRUN
NVIDIA_BIGNLP_VERSION= TRT_VER=10.5.0.18 CUDA_VER=12.6 CUDNN_VER=9.5.0.50-1 NCCL_VER= CUBLAS_VER=12.6.3.3-1 NVRTC_VER=12.6.77-1 REINSTALL_APEX=False APEX_COMMIT=73375b3bbcb59a5d6ff43f2fafd00b9ecdbe0417 MCORE_COMMIT=core_r0.10.0 SRC_DIR=/src/tensorrt_llm CPP_BUILD_DIR=/src/tensorrt_llm/cpp/build TARGET_ARCH=x86 INSTALL_VLLM=True ALIGNER_COMMIT=r0.6.0 LAUNCHER_COMMIT=5730fac6f97795931325cab0ac5dce1924cdcb3f DATA_CURATOR_COMMIT=r0.6.0 INSTALL_DATA_CURATOR=True MODELOPT_VERSION=0.19.0 NEMO_COMMIT=r2.1.0 RESIL_COMMIT=97aad77609d2e25ed38ac5c99f0c13f93c48464e NEMO_RUN_COMMIT=b4e2258f61b88c53b77996b5f9ed871ee666d85f PRE_COMPILE_QUANT_KERNELS=True chmod -R a+w /workspace
12/21/2024 7:06 PM UTC
...