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.

  • 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
    ...

    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.