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
0e035c40b5812cfc3f209e3726ad5e7d2c76f81353e84f71e2fdb153f0a86535CONFIG
Entrypoint /usr/local/bin/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
10/01/2021 9:08 PM UTC
b93f18f41567206a61f43a0b3d84bc932671ff7470c38b0f8cf6f73e86ec3f60COPY
tutorials /workspace/nemo/tutorials
10/01/2021 9:08 PM UTC
f42858d1acf638dcca5f0f9242d6444f333995705590e2e1019a14d906a0a631COPY
tests /workspace/nemo/tests
10/01/2021 9:08 PM UTC
2e015f64e1ac9ec2c1af4b1e1cff3b987ee92c3dfe2637fa6457bb9f50517ebaCOPY
examples /workspace/nemo/examples
10/01/2021 9:08 PM UTC
a712d72091ee185ca8aa4c42229bc29b5549307485c09e6cc72a4156a73de1a4COPY
scripts /workspace/nemo/scripts
10/01/2021 9:08 PM UTC
8217ac782e6e30aba88bae862b8ff1d4861d1bb85e3b0578344a50485b03d969WORKDIR
/workspace/nemo
10/01/2021 9:08 PM UTC
e343a77dd0b2adb468ff00d21a84e9c0bc1dbc8dc4edf897f0904ad7d6484a9bRUN
NEMO_VERSION=1.4.0 conda install -c numba numba=0.53.1
10/01/2021 9:08 PM UTC
a52f4f6f6f7df178c2c15e6741193bc8e4a98ab0de80474fe77cc0959dfb8722RUN
NEMO_VERSION=1.4.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"
10/01/2021 9:07 PM UTC
1680bda756d07a421336bded985b0c69cafa0b6a658edfcce8b64013e4d54065RUN
NEMO_VERSION=1.4.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
10/01/2021 9:07 PM UTC
3539f92a12286e74efa37d91bdb7db0892b1699205b99689b9e1f5fb369795f3ARG
NEMO_VERSION=1.4.0
10/01/2021 9:07 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.