docs: update wheel install instructions#260
Conversation
mckornfield
commented
Mar 18, 2026
- Update docs pages as appropriate for install instructions
There was a problem hiding this comment.
Pull request overview
Updates the user-guide documentation to provide complete wheel installation commands (including required package indexes) and to centralize install guidance by linking troubleshooting steps back to the Installation section.
Changes:
- Expand Getting Started → Install the Package with pip/uv commands and required index URLs for CUDA 12.8 and CPU installs.
- Add uv-specific guidance (
--index-strategy unsafe-best-match) for multi-index resolution. - Update Troubleshooting (“No GPU Detected”) to link to the canonical Installation instructions instead of embedding a single command.
Reviewed changes
Copilot reviewed 2 out of 2 changed files in this pull request and generated 3 comments.
| File | Description |
|---|---|
| docs/user-guide/troubleshooting.md | Replaces an inline CUDA install command with a link to the central Installation section. |
| docs/user-guide/getting-started.md | Adds detailed pip/uv install commands including index configuration for CUDA/CPU/bare installs. |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
|
we should update this for |
There was a problem hiding this comment.
Pull request overview
Updates installation documentation to include the correct index URLs needed for CUDA/CPU extras (PyTorch + FlashInfer) and aligns dependency resolution metadata for FlashInfer in uv configuration/lock.
Changes:
- Document
pip/uvinstall commands for CUDA 12.8 and CPU variants, including required index URLs. - Add
tool.uv.sourcesentries forflashinfer-pythonandflashinfer-cubin, and updateuv.lockto resolve cu128 FlashInfer deps from the FlashInfer index. - Update troubleshooting guidance and the tutorial notebook install snippet to reference the new install approach.
Reviewed changes
Copilot reviewed 4 out of 5 changed files in this pull request and generated 5 comments.
Show a summary per file
| File | Description |
|---|---|
uv.lock |
Adds sdist metadata for griffelib and pins FlashInfer cu128 deps to the FlashInfer index with conflicts metadata. |
pyproject.toml |
Configures uv sources for flashinfer-python / flashinfer-cubin to resolve from the FlashInfer cu128 index for the cu128 extra. |
docs/user-guide/troubleshooting.md |
Points users to the central install instructions (including index URLs) instead of embedding a partial command. |
docs/user-guide/getting-started.md |
Expands installation section with explicit pip/uv commands and rationale for uv index priority behavior. |
docs/tutorials/safe-synthesizer-101.ipynb |
Updates the Colab install cell to use public indexes (PyTorch + FlashInfer) instead of the internal Artifactory endpoint. |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
There was a problem hiding this comment.
Pull request overview
Copilot reviewed 4 out of 5 changed files in this pull request and generated 4 comments.
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
b18e003 to
7acc2cb
Compare
7acc2cb to
23bf7a6
Compare
There was a problem hiding this comment.
Pull request overview
Copilot reviewed 4 out of 5 changed files in this pull request and generated 2 comments.
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
| } | ||
| }, | ||
| "outputs": [], | ||
| "source": "%%capture\n!uv pip install nemo-safe-synthesizer[engine,cu128] --index https://flashinfer.ai/whl/cu128 --index https://download.pytorch.org/whl/cu128 --index-strategy unsafe-best-match\n!uv pip install datasets\n" |
There was a problem hiding this comment.
The shell command uses an unquoted extras specifier (nemo-safe-synthesizer[engine,cu128]). In many shells, [...] is treated as a glob character class, which can cause the requirement to be rewritten or fail depending on shell settings. Quote the requirement string to make the command robust.
| "source": "%%capture\n!uv pip install nemo-safe-synthesizer[engine,cu128] --index https://flashinfer.ai/whl/cu128 --index https://download.pytorch.org/whl/cu128 --index-strategy unsafe-best-match\n!uv pip install datasets\n" | |
| "source": "%%capture\n!uv pip install \"nemo-safe-synthesizer[engine,cu128]\" --index https://flashinfer.ai/whl/cu128 --index https://download.pytorch.org/whl/cu128 --index-strategy unsafe-best-match\n!uv pip install datasets\n" |
Signed-off-by: Matt Kornfield <mkornfield@nvidia.com>
23bf7a6 to
b760b61
Compare
There was a problem hiding this comment.
Pull request overview
Copilot reviewed 4 out of 5 changed files in this pull request and generated 1 comment.
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
| "# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.\n", | ||
| "# SPDX-License-Identifier: Apache-2.0\n", | ||
| "\n", | ||
| "#TODO: Replace the internal NVIDIA Artifactory endpoint with the public PyPI release when available.\n", | ||
| "!uv pip install nemo-safe-synthesizer[engine,cu128] --extra-index-url \"https://urm.nvidia.com/artifactory/api/pypi/nv-shared-pypi-local/simple\"\n", | ||
| "%%capture\n", | ||
| "!uv pip install \"nemo-safe-synthesizer[engine,cu128]\" --index https://flashinfer.ai/whl/cu128 --index https://download.pytorch.org/whl/cu128 --index-strategy unsafe-best-match\n", |
There was a problem hiding this comment.
In this notebook cell, %%capture is placed after comment lines. IPython requires cell magics (%%...) to be the very first line in the cell; otherwise the cell will error at runtime. Move %%capture to the top of the source list (before the SPDX comments), or drop the magic and use an alternative output-suppression approach.
There was a problem hiding this comment.
The %%capture will cause an error if it's not on the first time, hence why we have %%capture and then the SPDX comments.
| "# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.\n", | ||
| "# SPDX-License-Identifier: Apache-2.0\n", | ||
| "\n", | ||
| "#TODO: Replace the internal NVIDIA Artifactory endpoint with the public PyPI release when available.\n", | ||
| "!uv pip install nemo-safe-synthesizer[engine,cu128] --extra-index-url \"https://urm.nvidia.com/artifactory/api/pypi/nv-shared-pypi-local/simple\"\n", | ||
| "%%capture\n", | ||
| "!uv pip install \"nemo-safe-synthesizer[engine,cu128]\" --index https://flashinfer.ai/whl/cu128 --index https://download.pytorch.org/whl/cu128 --index-strategy unsafe-best-match\n", |
There was a problem hiding this comment.
The %%capture will cause an error if it's not on the first time, hence why we have %%capture and then the SPDX comments.
Signed-off-by: Matt Kornfield <mkornfield@nvidia.com>
775b0d9 to
46650ca
Compare
* Update docs/notebook pages as appropriate for install instructions --------- Signed-off-by: Matt Kornfield <mkornfield@nvidia.com>