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
d34131a578996773494f69753bea197db5127038231205dbdfae2e864cef3f2fCONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
07/30/2021 10:51 PM UTC
098c90dfa98693a2a1f62ea86763ba783f995b76467cacf25b00b1ed83204046COPY
tutorials /workspace/nemo/tutorials
07/30/2021 10:51 PM UTC
647442733b293888832c381290e35af019fb282adc12a05a4c629be95046547bCOPY
tests /workspace/nemo/tests
07/30/2021 10:51 PM UTC
75ee5c881180cf0c0000b0b188cdb634169f77183c1013736e2be4a554e8fbe4COPY
examples /workspace/nemo/examples
07/30/2021 10:51 PM UTC
427289945733283319eee838b4a0c4cb49bca194cdf0f40b629b936c6b98d9d4COPY
scripts /workspace/nemo/scripts
07/30/2021 10:51 PM UTC
dfd81d9ead4085c4c7f711f1c224f9817edf67c6d2593b2d671355642a64aec2WORKDIR
/workspace/nemo
07/30/2021 10:51 PM UTC
d713e307ad34b08be080ea2660499d019fe4386686b33873bd0f87e0cc0fd97bRUN
NEMO_VERSION=1.0.2 conda install -c numba numba=0.53.1
07/30/2021 10:51 PM UTC
4dff85c1395a92496915b2e4af8263bcc5c814596078b9f6fb21806985b1b638RUN
NEMO_VERSION=1.0.2 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"
07/30/2021 10:50 PM UTC
241b1c53b6e8fd2673f52e36b478e38b8e3a359d440d975ff3cd4b7e5c3466b0RUN
NEMO_VERSION=1.0.2 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
07/30/2021 10:50 PM UTC
b530bd2a7abe624997b1e8c3c39117f2706ce2c0fc431fa0a2ce344863cebd51ARG
NEMO_VERSION=1.0.2
07/30/2021 10:50 PM UTC
...