GraphCanon updated today · GitHub synced today
Trust & integrity
Full report- Maintenance
- Very active (0d since push)
- As of today · Source: github_public_v1
- Provenance
- Not a fork · Organization account
- As of today · Source: github_public_v1
- Security (OSV)
- No lockfile
- As of today · Source: none
Public GitHub metadata and optional OSV dependency scans. Signals, not a guarantee. Trust methodology.
Overview
NVIDIA-NeMo/Speech is a comprehensive toolkit for Automatic Speech Recognition (ASR), speaker diarization, speaker recognition, speech synthesis (TTS), and more. It is built on PyTorch and supports CUDA for efficient GPU utilization.
Capability facts
- Languages
- python
Source: github.language+pyproject.toml · Jul 11, 2026
Categories
Compatibility
Sourced claims from the README excerpt - not unsourced marketing copy.
Source: README excerpt (regex_v1, Jul 11, 2026)
NeMo Speech works with the **Python, PyTorch, and CUDA versions of your choosing**:Source link
Tags
README
Requirements
NeMo Speech works with the Python, PyTorch, and CUDA versions of your choosing:
- Python 3.12 or above
- PyTorch 2.7 or above (CPU, CUDA, etc. — your choice)
- NVIDIA GPU + CUDA (required for training; recommended for inference)
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.12, CUDA 12.6/13.2 — are simply the combination 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.
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 also installs over your existing environment via pip — see the pip fallback below.
Docker (turnkey, our supported stack)
NGC container: Coming soon — the pull command for the prebuilt NeMo Speech container image will be published here.
To build the container from source (CUDA 13 / H100+ by default):
git clone https://github.com/NVIDIA-NeMo/NeMo.git
cd NeMo
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).