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
5b5f74560c8f3f87c8de5bf1a459450aa02d0a899784d5db7b4f50f87e878ce9CONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
09/03/2021 10:45 PM UTC
b6c516bfc918237f4552ba7fae572e8c67e23ed1d2ab283b357fb6128965045eCOPY
tutorials /workspace/nemo/tutorials
09/03/2021 10:45 PM UTC
bd5898de074e0c74c3e41813ed7a88cc43cb9267433db0e060a93f113f8d0d10COPY
tests /workspace/nemo/tests
09/03/2021 10:45 PM UTC
5805be95b0e4f8f21562f46f7a9f5c2f0e48b458589d8fdf1fc74b60b08d8d76COPY
examples /workspace/nemo/examples
09/03/2021 10:45 PM UTC
f62dd7711eace6cfae568d5152fc38af428428c5d77acc3491a958eff3d171f0COPY
scripts /workspace/nemo/scripts
09/03/2021 10:45 PM UTC
dfe31a398673559ec83959950cf818fdb2040cd7d5875f275592407bf6e2fd50WORKDIR
/workspace/nemo
09/03/2021 10:45 PM UTC
b68e9f5ba8a8ec9e0a8089bb7d347c001351704a3035d25f7222b4c352f8d306RUN
NEMO_VERSION=1.3.0 conda install -c numba numba=0.53.1
09/03/2021 10:45 PM UTC
63230f4a2879215a8e07c0c7ae43484299821be4b32f6919cc99fdd54eb454c0RUN
NEMO_VERSION=1.3.0 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"
09/03/2021 10:44 PM UTC
f37789c1a2fdddb7bb37cf49fe316d93319ae729a382d392b1c84a1959f431efRUN
NEMO_VERSION=1.3.0 /usr/bin/test -n "$NEMO_VERSION" &&
  /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc &&
  /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc
09/03/2021 10:44 PM UTC
6d4fc08671fa12fb256e0e60ff6b02dc75ed49f76457c10daad62a74774c689cARG
NEMO_VERSION=1.3.0
09/03/2021 10:44 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.