You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: CLAUDE.md
+1-1Lines changed: 1 addition & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -10,7 +10,7 @@ NeMo Speech — toolkit for training/deploying speech models (ASR, TTS, Speech L
10
10
11
11
See the canonical installation guide — [`docs/source/starthere/install.rst`](docs/source/starthere/install.rst) (published at https://docs.nvidia.com/nemo/speech/nightly/) — for the uv, pip (bring-your-own Python/PyTorch/CUDA), Docker, and optional `compiled` (SpeechLM2/Automodel) install paths.
12
12
13
-
Dev quickstart: `uv sync --extra all --extra cu13` (Python 3.10+, PyTorch 2.6+; `test`/`docs` are `--group`s, not extras).
13
+
Dev quickstart: `uv sync --extra all --extra cu13` (Python 3.12+, PyTorch 2.7+; `test`/`docs` are `--group`s, not extras).
Copy file name to clipboardExpand all lines: README.md
+4-4Lines changed: 4 additions & 4 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -51,11 +51,11 @@ For technical documentation, please see the
51
51
52
52
NeMo Speech works with the **Python, PyTorch, and CUDA versions of your choosing**:
53
53
54
-
- Python 3.10 or above
55
-
- PyTorch 2.6 or above (CPU, CUDA, etc. — your choice)
54
+
- Python 3.12 or above
55
+
- PyTorch 2.7 or above (CPU, CUDA, etc. — your choice)
56
56
- NVIDIA GPU + CUDA (required for training; recommended for inference)
57
57
58
-
If you already have a Python/PyTorch/CUDA stack, NeMo Speech installs on top of it **without replacing it** — the `nemo-toolkit` package only requires `torch>=2.6`, 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.
58
+
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.
59
59
60
60
As of [Pytorch 2.6](https://docs.pytorch.org/docs/stable/notes/serialization.html#torch-load-with-weights-only-true),
61
61
`torch.load` defaults to using `weights_only=True`. Some model checkpoints may require using `weights_only=False`.
@@ -103,7 +103,7 @@ For A100, set `GPU_TARGET=a100` — A100 works with **both CUDA 12 and CUDA 13**
103
103
104
104
### From PyPI with pip (fallback — bring your own versions)
105
105
106
-
Prefer your own Python/PyTorch/CUDA? `nemo-toolkit` only requires `torch>=2.6`, so install your PyTorch first (any version ≥ 2.6 for your CPU/CUDA/etc. target — see the [PyTorch install matrix](https://pytorch.org/get-started/locally/)), then add NeMo and it **keeps your build**. `uv pip` (uv's fast, pip-compatible installer) works like `pip`:
106
+
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](https://pytorch.org/get-started/locally/)), then add NeMo and it **keeps your build**. `uv pip` (uv's fast, pip-compatible installer) works like `pip`:
107
107
108
108
```bash
109
109
uv pip install 'nemo-toolkit[asr,tts]'# or plain: pip install 'nemo-toolkit[asr,tts]'
Copy file name to clipboardExpand all lines: docs/source/starthere/install.rst
+6-6Lines changed: 6 additions & 6 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -10,15 +10,15 @@ Prerequisites
10
10
11
11
NeMo Speech works with the **Python, PyTorch, and CUDA versions of your choosing**:
12
12
13
-
#. **Python** 3.10 or above
14
-
#. **PyTorch** 2.6 or above, for your chosen target (CPU, CUDA, etc.)
13
+
#. **Python** 3.12 or above
14
+
#. **PyTorch** 2.7 or above, for your chosen target (CPU, CUDA, etc.)
15
15
#. **NVIDIA GPU + CUDA** (required for training; CPU-only inference is possible but slow)
16
16
#. **uv** for the fastest source/PyPI workflow (``pip`` also works in a prepared environment)
17
17
18
18
.. admonition:: Bring your own Python / PyTorch / CUDA
19
19
:class: important
20
20
21
-
The recommended install path is uv (below), which gives you our actively-tested stack. But NeMo Speech can also install *on top of* an existing environment: the ``nemo-toolkit`` package only requires ``torch>=2.6``, so if you already have a Python, PyTorch, and CUDA stack, your pre-installed PyTorch is **kept, not replaced** (see :ref:`the pip fallback <install-from-pypi>`).
21
+
The recommended install path is uv (below), which gives you our actively-tested stack. But NeMo Speech can also install *on top of* an existing environment: if you already have a Python, PyTorch, and CUDA stack that satisfies the minimums above, your pre-installed PyTorch is **kept, not replaced** (see :ref:`the pip fallback <install-from-pypi>`).
22
22
23
23
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.
24
24
@@ -157,11 +157,11 @@ See the header of ``docker/Dockerfile`` for all build arguments (``BASE_IMAGE``,
157
157
Install from PyPI with pip (fallback — bring your own versions)
Prefer your own Python/PyTorch/CUDA? Install your preferred PyTorch first (any version ≥ 2.6 for your CPU/CUDA/etc. target — see `PyTorch's install matrix <https://pytorch.org/get-started/locally/>`_), then add NeMo. Because ``nemo-toolkit`` only requires ``torch>=2.6``, your pre-installed PyTorch is kept, not replaced. ``uv pip`` (uv's fast, pip-compatible installer) works just like ``pip``:
160
+
Prefer your own Python/PyTorch/CUDA? Install your preferred PyTorch first (any version ≥ 2.7 for your CPU/CUDA/etc. target — see `PyTorch's install matrix <https://pytorch.org/get-started/locally/>`_), then add NeMo. Your pre-installed PyTorch is kept, not replaced. ``uv pip`` (uv's fast, pip-compatible installer) works just like ``pip``:
161
161
162
162
.. code-block:: bash
163
163
164
-
uv venv --python 3.12 # any Python >= 3.10 your PyTorch supports — or use your own env
164
+
uv venv --python 3.12 # any Python >= 3.12 your PyTorch supports — or use your own env
165
165
source .venv/bin/activate
166
166
167
167
# 1) Your choice of PyTorch (example: CUDA 12.6 build). Skip if you already have one.
@@ -183,7 +183,7 @@ To instead have the installer pull *our* pinned PyTorch build, add the matching
183
183
184
184
.. tip::
185
185
186
-
Prefer a conda environment? Create and activate one (``conda create -n nemo python=3.10 -y && conda activate nemo``), then run the same ``uv`` or ``pip`` commands above inside it. NeMo Speech does not require a separate conda CUDA toolkit.
186
+
Prefer a conda environment? Create and activate one (``conda create -n nemo python=3.12 -y && conda activate nemo``), then run the same ``uv`` or ``pip`` commands above inside it. NeMo Speech does not require a separate conda CUDA toolkit.
0 commit comments