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
5e7a70a161f2cd2ecab8ab8ef11f5bb8a384e3c19e78ada36f8b37aed481a6d3CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace/nemo; ExposedPorts 6006/tcp, 8888/tcp
06/19/2023 7:06 PM UTC
b7655f2641ff1eca0677236d2362923cfa46e6ced5de23ed0a07389e0e3da35fRUN
NEMO_VERSION=1.19.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh &&
  chmod +x start-jupyter.sh
06/19/2023 7:06 PM UTC
b33de2b50ec0c248a9d7c0411d099ffe40c9c44d26615b79078e8fb8d7c1748dCOPY
tutorials /workspace/nemo/tutorials
06/19/2023 7:06 PM UTC
913e62d07a7776668a43c5ee9ba838abe7b7547457609034ee24ad3758886e86COPY
tests /workspace/nemo/tests
06/19/2023 7:06 PM UTC
f31cd10b1e0716cc2c56e475429b1a952054b5d4de1bf2214db8c68c0dc4ec3cCOPY
examples /workspace/nemo/examples
06/19/2023 7:06 PM UTC
9a0786e9f56db05ab6948422f8f11f94d93caa4bb75f9b63c8c57197b610f402COPY
scripts /workspace/nemo/scripts
06/19/2023 7:06 PM UTC
9a5af4afa83342d2a6c7f580ee4fb58a97f05c8bfe5c56f8b4bc210f172a0b16WORKDIR
/workspace/nemo
06/19/2023 7:06 PM UTC
df3c75a3302e1de910f01d733be15e6975b37a01cba77db40af4c55628cb5da9RUN
NEMO_VERSION=1.19.0 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"
06/19/2023 7:06 PM UTC
b4a6d3ea7c23ebc87dd4c92d571ecfebd1e7f5d197a23a03faf430597e0d4042RUN
NEMO_VERSION=1.19.0 cd /tmp/nemo &&
  pip install ".[all]"
06/19/2023 7:05 PM UTC
d464af919d23dca4174eda3097c4c0cd1e22af7d3505a03362983a8cac936ad9RUN
NEMO_VERSION=1.19.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
06/19/2023 7:05 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.