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
487c2d38ffeda626a2fe874b4a6877fc9593f91fba4a513a919f678a506b2772CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
04/01/2024 7:22 PM UTC
849db964633d6aeaf6d147b5b5c31c21f07106f3f27f85cc3c509a0a6f692042ARG
NVIDIA_BUILD_REF
04/01/2024 7:22 PM UTC
43a1e2fffee3c98a4594c8754a036446b7ef7a1ff97130e0401c552126a1daf3LABEL
com.nvidia.build.id=82611821
04/01/2024 7:22 PM UTC
6ac1e91d8edde580fff31a11f4f08fbcff87cc614ade7f7084dfab733fd7d61cENV
NVIDIA_BUILD_ID=82611821
04/01/2024 7:22 PM UTC
dcee4528b2fd6f24175a467ad0fc305f91888e179d763423765e3bba984ed6ecARG
NVIDIA_BUILD_ID
04/01/2024 7:22 PM UTC
3fc07cbbfc735e24f8ee94846a625231132b53c72adadad536bf67005a25ff52RUN
chmod -R a+w /workspace
04/01/2024 7:22 PM UTC
2b1d6d22d79118ae47d7fdff80d0a15aad67be75b11ff2dd3ab1842c6e4ca74cWORKDIR
/workspace
04/01/2024 7:22 PM UTC
55dd4709dadacb1c037937dc63d8bdd7b4cb7fe239d854b2707f5413b36865e1RUN
TCNN_CUDA_ARCHITECTURES="52,60,61,70,75,80,86,89,90" pip install --no-cache-dir git+https://github.com/NVlabs/tiny-cuda-nn@6f018a9cd1b369bcb247e1d539968db8e48b2b3f#subdirectory=bindings/torch
04/01/2024 7:22 PM UTC
f9c35dc15a6c33f096497605aa7972bda9f1fc5e2a671cdddf1f485d70a5bc0aRUN
cd /tmp &&
  git clone https://github.com/ashawkey/stable-dreamfusion.git &&
  cd stable-dreamfusion &&
  git checkout 5550b91862a3af7842bb04875b7f1211e5095a63 &&
  find . -type f -name 'setup.py' -exec sed -i 's/c++14/c++17/g' {} + &&
  pip install --no-cache-dir ./raymarching &&
  pip install --no-cache-dir ./shencoder &&
  pip install --no-cache-dir ./freqencoder &&
  TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6 9.0+PTX" pip install --no-cache-dir ./gridencoder &&
  cd /opt &&
  rm -rf /tmp/stable-dreamfusion
04/01/2024 6:35 PM UTC
c9e8c578bd587cba897c05f38a158fcab9ec6f28f0a5a4b8306a24d9cc85a729RUN
pip install --no-cache-dir git+https://github.com/NVlabs/nvdiffrast.git@c5caf7bdb8a2448acc491a9faa47753972edd380
04/01/2024 6:08 PM 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.