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
7c4d7ac14158d777c1de26bd799b9e72b91856a91be8c46383d00e132c97c8b8CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
02/03/2022 11:32 PM UTC
e377ad9569f83a827a72b2ee7268c59a09b179cac66b231977df400213238d0eCOPY
tutorials /workspace/nemo/tutorials
02/03/2022 11:32 PM UTC
f98c30f01ab907ca59f2e57bae970cfd90f62f5e37f8896e7940f87a9d6f6562COPY
tests /workspace/nemo/tests
02/03/2022 11:32 PM UTC
52179f839434771391f611d16bd50a347fd118a77e03ed597423889ca1b738fcCOPY
examples /workspace/nemo/examples
02/03/2022 11:32 PM UTC
e47630f5396fb8370741e9467348c4a230dbbf4d388c2981976dd47b254f4e64COPY
scripts /workspace/nemo/scripts
02/03/2022 11:32 PM UTC
11a8a076fa8963f25f05f19a01ee5d879d22c475a8869f510d60423e2e382f03WORKDIR
/workspace/nemo
02/03/2022 11:32 PM UTC
850fc14893ac0299763552fa9aa5857f35db94cbbf503e00a6de9ac36efed5d8RUN
NEMO_VERSION=1.6.1 rm -rf /opt/conda/pkgs/conda-4.11.0-py38h578d9bd_0/info/test/tests/data/env_metadata/
02/03/2022 11:32 PM UTC
41d7d686468f6f4742e30b3b03796155b13c7f6f6b86d3e5fae4cad0d1229ad3RUN
NEMO_VERSION=1.6.1 conda install -c numba numba=0.54.1
02/03/2022 11:32 PM UTC
6316b1a7e3037a4c32885e459faa79a64cb41e8a328f562c31db8c211c673442RUN
NEMO_VERSION=1.6.1 cd /tmp/nemo &&
  pip install ".[all]" &&
  python -c "import nemo.collections.asr as nemo_asr" &&
  python -c "import nemo.collections.nlp as nemo_nlp" &&
  python -c "import nemo.collections.tts as nemo_tts" &&
  python -c "import nemo_text_processing.text_normalization as text_normalization"
02/03/2022 11:31 PM UTC
ac4c0f214f4ac898069aea78a4a07832baac80d85868c72fd7f0894b237f3a7fRUN
NEMO_VERSION=1.6.1 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
02/02/2022 9:03 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.