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
00b9c6242d8514d2741f866fde15122a8e5039ad2bef290709b0ddc9665ffcddCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
05/29/2024 10:33 AM UTC
783f81f6ffa1c3eacaba3f1180f49ad4689849276ca50ca49b23b8b3d3417946ARG
NVIDIA_BUILD_REF
05/29/2024 10:33 AM UTC
f9b93586c9de0597054ed6dc1d277e0933fa6d9a98a001325317866ea2c99125LABEL
com.nvidia.build.id=82611821
05/29/2024 10:33 AM UTC
23c887431343c39e77d2bd9428ce083c14cff7a30e3eed55a62206d73e1bb045ENV
NVIDIA_BUILD_ID=82611821
05/29/2024 10:33 AM UTC
ab16bd72c01e172bd873475d126870feb71dc4875bd4cc266c42fa3176d64644ARG
NVIDIA_BUILD_ID
05/29/2024 10:33 AM UTC
6362aa46e036aa5c92795759a766e603c285daacba7c074fd75901432497058cRUN
NEMO_COMMIT=r2.0.0rc0 chmod -R a+w /workspace
05/29/2024 10:33 AM UTC
eb642de9b201dc1e413e96ee222ae8f3c9cf7b4b4a6df8bb2d7add0349ef38c3WORKDIR
/workspace
05/29/2024 10:33 AM UTC
27c763be63da60a74de6da158a89c268a59df02f93ed2ade718db9eb41281823RUN
NEMO_COMMIT=r2.0.0rc0 pip uninstall -y causal-conv1d &&
  pip install --no-cache-dir 'causal-conv1d==1.2.0.post2'
05/29/2024 10:33 AM UTC
868360d72183d38f5ae23f4f6b115bb6a6aa9441e18b55d86ef237d18369cc07RUN
NEMO_COMMIT=r2.0.0rc0 pip uninstall -y jupyterlab aiohttp onnx &&
  pip install --no-cache-dir jupyterlab==3.6.7 aiohttp==3.9.5 onnx==1.16.0
05/29/2024 10:33 AM UTC
0a9e359d74ba0e3af3300b21f037d4abc00f2bd49a5afc4c4f74c6bb305ce93eRUN
NEMO_COMMIT=r2.0.0rc0 python -c "from transformers import CLIPImageProcessor; vit_p=CLIPImageProcessor.from_pretrained('openai/clip-vit-large-patch14'); vit_p_336=CLIPImageProcessor.from_pretrained('openai/clip-vit-large-patch14-336');"
05/29/2024 10:32 AM UTC
...

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.