Speech README

The project's README from github.com/NVIDIA-NeMo/Speech. As one markdown file: README.md.

Project Status: Active -- The project has reached a stable, usable state and is being actively developed. CodeQL NeMo Speech license Release version Python version PyPi total downloads Code style: black

NVIDIA NeMo Speech

Checkout our HuggingFace🤗 collection for the latest open weight checkpoints and demos!

Updates

NeMo Speech 3.0 is now available as release v3.0.0 and in the 26.07.00 NeMo Speech NGC container. The final NeMo release before the repository split was v2.7.3, available in the 26.04 NeMo NGC container.

Introduction

NVIDIA NeMo Speech is built for researchers and PyTorch developers working on Speech models including Automatic Speech Recognition (ASR), Text to Speech (TTS), and Speech LLMs. It is designed to help you efficiently create, customize, and deploy new AI models by leveraging existing code and pre-trained model checkpoints.

For technical documentation, please see the NeMo Speech Developer Documentation.

Requirements

NeMo Speech works with the Python, PyTorch, and CUDA versions of your choosing:

If you already have a Python/PyTorch/CUDA stack that satisfies those minimums, NeMo Speech installs on top of it without replacing it, so your existing PyTorch build is kept (see the install options below). The versions pinned in uv.lock and shipped in the official container — Python 3.13, PyTorch 2.11 with CUDA 12.9 or PyTorch 2.12 with CUDA 13.2 — are simply the combinations we actively test and support. They make setup turnkey and reproducible, but they are not a hard requirement.

As of Pytorch 2.6, torch.load defaults to using weights_only=True. Some model checkpoints may require using weights_only=False. In this case, you can set the env var TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD=1 before running code that uses torch.load. However, this should only be done with trusted files. Loading files from untrusted sources with more than weights only can have the risk of arbitrary code execution.

Developer Documentation

Version Description
3.0.0 (latest release) NeMo Speech 3.0.0 documentation
Nightly Documentation for the latest main branch

Install NeMo Speech

The recommended way to install NeMo Speech is from source with uv, which reproduces our actively-tested stack from the committed uv.lock. If you need different Python/PyTorch/CUDA versions, NeMo Speech also installs over your existing environment via pip — see the pip fallback below.

git clone https://github.com/NVIDIA-NeMo/Speech.git
cd Speech
uv sync --extra all --extra cu13     # CUDA 13.x (recommended) — use --extra cu12 for CUDA 12.x

This installs our supported stack (Python 3.13, PyTorch 2.12, CUDA 13.2) into .venv/ with NeMo Speech editable. Add --group test for the test suite or --group docs to build the docs; run tools via uv run <cmd> or activate with source .venv/bin/activate. On Linux, cu12 and cu13 are mutually exclusive — pass exactly one (cu13 is the default). For the exact container baseline, add --locked --python 3.13 (the path the Dockerfile and CI use).

SpeechLM2 / Automodel: the Automodel backend runs without any compiled dependencies. It can optionally benefit from dedicated accelerated backends (Transformer Engine, FlashAttention, Mamba, grouped-GEMM/MoE, DeepEP) for better performance — these source-built kernels come from the compiled (Hopper/Blackwell) or compiled-a100 (A100) extras, built by docker/Dockerfile (GPU_TARGET=h100plus / a100). See the installation guide for the full list and build details.

Docker (turnkey, our supported stack)

The latest prebuilt NeMo Speech image is the 26.07.00 NGC container:

docker pull nvcr.io/nvidia/nemo-speech:26.07.00
docker run --rm -it --gpus all -v "$PWD:/workspace" nvcr.io/nvidia/nemo-speech:26.07.00 bash

To build the container from source (CUDA 13 / H100+ by default):

git clone https://github.com/NVIDIA-NeMo/Speech.git
cd Speech
docker buildx build -f docker/Dockerfile -t nemo-speech .          # CUDA 13 / H100+ (default)
docker run --rm -it --gpus all -v "$PWD:/workspace" nemo-speech bash

For A100, set GPU_TARGET=a100 — A100 works with both CUDA 12 and CUDA 13 (CUDA 13, the default base image, is recommended; the CUDA 12 base is a convenience). See the header of docker/Dockerfile for all build arguments (BASE_IMAGE, GPU_TARGET).

From PyPI with pip (fallback — bring your own versions)

Prefer your own Python/PyTorch/CUDA? Install your PyTorch first (any version ≥ 2.7 for your CPU/CUDA/etc. target — see the PyTorch install matrix), then add NeMo Speech and it keeps your build. uv pip (uv's fast, pip-compatible installer) works like pip:

uv pip install 'nemo-toolkit[asr,tts]'   # or plain: pip install 'nemo-toolkit[asr,tts]'

⚠️ Do not use uv sync --locked for a bring-your-own stack — it applies uv.lock and replaces your Python/PyTorch/CUDA with the supported baseline. Use uv pip/pip here; reserve uv sync --locked for reproducing our stack.

To instead pull our pinned PyTorch build, add the CUDA extra and the matching wheel index (pip/uv pip do not read uv's project index config, so --extra-index-url is required):

pip install 'nemo-toolkit[asr,tts,cu13]' --extra-index-url https://download.pytorch.org/whl/cu132   # CUDA 13.x
pip install 'nemo-toolkit[asr,tts,cu12]' --extra-index-url https://download.pytorch.org/whl/cu129   # CUDA 12.x

Contribute to NeMo Speech

We welcome community contributions! Please refer to CONTRIBUTING.md for the process.

Licenses

NeMo Speech is licensed under the Apache License 2.0.

Speech (NVIDIA-NeMo/Speech)

Searching Database…

18.6K ·GitHub
这是什么?NVIDIA 的一个开源框架,用于构建语音 AI 模型——自动语音识别(ASR)、语音合成(TTS)和语音大语言模型,附带可自定义、可部署的预训练模型权重。
它是一个框架,还是一组现成的模型?两者都是:它既是一个用于创建、定制和部署模型的 PyTorch 框架,也通过 HuggingFace 集合提供开放权重的模型,另外还有带演示的 NGC 容器。
我怎么把它跑起来?推荐方式:用 uv 从源码安装(能复现经过测试的 Python 3.13、PyTorch 2.12、CUDA 13.2 组合)。备选方式:在自带的 PyTorch 2.7+ 上用 pip install 安装,或者使用 26.07.00 的 NGC Docker 容器。
我需要 NVIDIA 的 GPU 吗?最低要求是 Python 3.12+ 和 PyTorch 2.7+。训练必须有带 CUDA 的 NVIDIA GPU,推理也建议使用;你现有的 PyTorch 安装会被保留,不会被替换。
安装时有什么坑要留意?在 Linux 上,cu12 和 cu13 这两个 extras 互斥——只能选其中一个传入。另外在自带依赖的环境里避免使用 'uv sync --locked':它会把你的 Python/PyTorch/CUDA 替换成锁定的基线版本。
流式 ASR 的延迟最低能到多少?Nemotron-3.5-ASR-Streaming-0.6B 支持 40 种语言,延迟可在 80 毫秒到 1 秒之间调节,单张 H100 上支持 240 到 2400 路并发流,基于缓存感知的 Fastconformer 架构。
它能用来做实时语音对话吗?可以——Nemotron 3 VoiceChat(抢先体验版)支持全双工、自然、可随时打断的低延迟对话,底层是 Nemotron Nano v2 大语言模型,配合 Nemotron 语音和 TTS 解码器。
MagpieTTS 模型支持哪些语言?MagpieTTS v2607 支持 12 种语言:英语、西班牙语、德语、法语、越南语、意大利语、中文、印地语和日语,另外新增了阿拉伯语、韩语和葡萄牙语。
听说 NeMo 以前也覆盖大语言模型,后来怎么了?2026 年这个仓库转向专注于音频、语音和多模态大语言模型。拆分前最后一个支持其他模态的版本是 v2.7.3。
Size
520.4 MB
Stars
18.6K
Age
7y
Commits · 90 Days