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
sha256:d847fd99b38ecba3a116909ce5632a630c7cec1541a2d8c5e06cb057cd61300fCONFIG
—
—
647f0559ab3fe9f8a95ba510c759668fb60fa43f853f87b6cf8e0a6209eba0dbLABEL
com.nvidia.build.ref=
04/22/2024 9:55 AM UTC
1a33a3767e9d9366337cdad660da58fd136e908a66c47c09be70db0d91af6534ARG
NVIDIA_BUILD_REF
04/22/2024 9:55 AM UTC
b326689ce0ae43e9a381f18bb695c0714c6f124ae2082d412e0a78ff2b4dce8cLABEL
com.nvidia.build.id=82611821
04/22/2024 9:55 AM UTC
0e66c9b2b4f41a84505c635cfc9d3310367504737f29084e23086622aa4ee7c2ENV
NVIDIA_BUILD_ID=82611821
04/22/2024 9:55 AM UTC
15c2b229eb64b32b47de0b845e0426bfdda57f596e27a8615e50df594b5bc917ARG
NVIDIA_BUILD_ID
04/22/2024 9:55 AM UTC
850792bd96da9c386b5b7072e36fdf73ed9fb85e7dc92782474725c20d7f5442RUN
chmod -R a+w /workspace
04/22/2024 9:55 AM UTC
8f828ff6300baf587c7c705281d558e633ea88dded123dcb50f14fb2b696bd61WORKDIR
/workspace
04/22/2024 9:55 AM UTC
9c6c0d9654695e7d385b78cf9d48270f76db0d5feec164b349efcb0a7ae6657dRUN
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/22/2024 9:55 AM UTC
c2f1f129e4b6bfc9d782c45a6cd91615f2cdacc664ec5d1158aa273718895389RUN
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/22/2024 9:04 AM UTC
...