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:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.ref=
02/18/2024 10:40 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_REF
02/18/2024 10:40 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.id=80741402
02/18/2024 10:40 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NVIDIA_BUILD_ID=80741402
02/18/2024 10:40 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_ID
02/18/2024 10:40 PM UTC
sha256:c065e287b6a1b63df8cc04d7a1cba8723af0d7ef015224136ecff7b3e2ce48e4RUN
chmod -R a+w /workspace
02/18/2024 10:40 PM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1WORKDIR
/workspace
02/18/2024 10:40 PM UTC
sha256:81f86ff59330a977757799eb6c680d833219869121b98136abb52887dd2fc1b1RUN
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
02/18/2024 10:40 PM UTC
sha256:8c69eb3668e346f95573dd99c3c8adf0222367929357ddc018c76ef1a167a436RUN
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
02/18/2024 9:53 PM UTC
sha256:6097cd6181e9eb05db9867ef7aa516da9e9850c36000387d62c98fe1cdc6322aRUN
pip install --no-cache-dir tritonclient==2.38.0 kornia==0.6 Pillow==9.3.0 webdataset==0.2.48 git+https://github.com/openai/CLIP.git@main#egg=clip -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers --src /opt accelerate seaborn datasets prefetch_generator diffusers==0.19.3 torchdiffeq torchsde jiwer editdistance pyannote.core pyannote.metrics addict yapf basicsr open_clip_torch torch-ema trimesh PyMCubes xatlas imageio pymeshlab nerfacc==0.5.3 einops_exts git+https://github.com/NVlabs/nvdiffrast.git@c5caf7bdb8a2448acc491a9faa47753972edd380
02/18/2024 9:27 PM UTC
...