NVIDIA NeMo Developer Guide¶
NeMo is a library for easy training, building and manipulating of AI models. NeMo’s current focus is providing great experience for Conversational AI.
NeMo models can be trained on multi-GPU and multi-node, with or without Mixed Precision Many models in NeMo come with high-quality pre-trained checkpoints.
Requirements¶
NeMo’s main requirements are:
Python 3.6 or 3.7
Pytorch 1.6 or above
Installation¶
apt-get update && apt-get install -y libsndfile1 ffmpeg && pip install Cython && pip install nemo_toolkit[all]==version
We recommend using NVIDIA’s PyTorch container
docker run --gpus all -it --rm -v <nemo_github_folder>:/NeMo --shm-size=8g \
-p 8888:8888 -p 6006:6006 --ulimit memlock=-1 --ulimit \
stack=67108864 nvcr.io/nvidia/pytorch:20.09-py3