ci: Install CI dependencies from uv lock by chtruong814 · Pull Request #15697 · NVIDIA-NeMo/NeMo

chtruong814 · 2026-05-13T20:50:48Z

Summary

Replace the CI Docker base with CUDA 13.1.2 and install dependencies with uv from pyproject.toml/uv.lock
Remove TE/Megatron-Core wheel stages and install_dep.sh usage from docker/Dockerfile.ci
Force Linux torch resolution to torch 2.10.0+cu130 and keep nemo_automodel pinned to the requirements manifest commit

Testing

UV_CACHE_DIR=/private/tmp/nemo-uv-cache /private/tmp/nemo-uv-latest/bin/uv lock --check
DOCKER_BUILDKIT=1 /opt/homebrew/bin/docker-buildx build --progress=plain -f docker/Dockerfile.ci -t nemo-ci-uv-test --load . (uv install completed; local image load failed due Docker disk space)

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

chtruong814 · 2026-05-13T21:20:36Z

/ok to test 611aa64

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

chtruong814 · 2026-05-14T03:01:40Z

/ok to test fd0cce2

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Signed-off-by: Piotr Żelasko <pzelasko@nvidia.com>

copy-pr-bot · 2026-05-27T14:19:58Z

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

Signed-off-by: Piotr Żelasko <pzelasko@nvidia.com>

pzelasko · 2026-05-27T14:59:07Z

/ok to test 88de033

copy-pr-bot · 2026-05-27T14:59:10Z

/ok to test 88de033

@pzelasko, there was an error processing your request: E2

See the following link for more information: https://docs.gha-runners.nvidia.com/cpr/e/2/

pzelasko · 2026-05-27T14:59:22Z

/ok to test fa5e5f9

pzelasko · 2026-05-27T19:22:27Z

/ok to test 074a90a

pzelasko · 2026-05-27T21:35:54Z

/ok to test eca699c

pzelasko · 2026-05-28T00:42:17Z

/ok to test 1eca063

Use decoder vocab size when generating synthetic TDT transcript labels so duration outputs from the joint are not sampled as labels. Move CUDA graph compile exception types into cuda_python_utils per review feedback. Signed-off-by: Piotr Żelasko <pzelasko@nvidia.com>

pzelasko · 2026-05-28T14:18:20Z

/ok to test a360c2f

github-actions · 2026-05-28T16:30:53Z

[🤖]: Hi @chtruong814 👋,

We wanted to let you know that a CICD pipeline for this PR just finished successfully.

So it might be time to merge this PR or get some approvals.

pzelasko

LGTM!

artbataev

LGTM, thanks!

chtruong814 · 2026-05-29T01:14:30Z

/ok to test a4a9c66

github-actions · 2026-05-29T06:03:14Z

[🤖]: Hi @chtruong814 👋,

We wanted to let you know that a CICD pipeline for this PR just finished successfully.

So it might be time to merge this PR or get some approvals.

chtruong814 added 2 commits May 13, 2026 13:40

Move requirements metadata to pyproject

882084f

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Update CI Dockerfile to install with uv

2e93ef9

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

chtruong814 requested a review from a team as a code owner May 13, 2026 20:50

copy-pr-bot Bot temporarily deployed to public May 13, 2026 20:51 Inactive

copy-pr-bot Bot had a problem deploying to test May 13, 2026 20:52 Error

chtruong814 added 2 commits May 13, 2026 15:54

Remove legacy requirements files

0d53a71

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Remove legacy Docker dependency installer

fb05691

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

copy-pr-bot Bot temporarily deployed to public May 13, 2026 20:57 Inactive

chtruong814 changed the title ~~Install CI dependencies from uv lock~~ ci: Install CI dependencies from uv lock May 13, 2026

copy-pr-bot Bot temporarily deployed to test May 13, 2026 20:58 Inactive

copy-pr-bot Bot temporarily deployed to public May 13, 2026 21:00 Inactive

copy-pr-bot Bot temporarily deployed to public May 13, 2026 21:01 Inactive

copy-pr-bot Bot temporarily deployed to public May 13, 2026 21:05 Inactive

chtruong814 added the Run e2e nightly label May 13, 2026

Fix container build

611aa64

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

github-actions Bot added the CI label May 13, 2026

copy-pr-bot Bot temporarily deployed to public May 13, 2026 21:21 Inactive

copy-pr-bot Bot temporarily deployed to test May 13, 2026 21:22 Inactive

copy-pr-bot Bot temporarily deployed to public May 13, 2026 21:24 Inactive

copy-pr-bot Bot temporarily deployed to public May 13, 2026 21:25 Inactive

copy-pr-bot Bot temporarily deployed to public May 13, 2026 21:28 Inactive

Fix nemo install

fd0cce2

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

copy-pr-bot Bot temporarily deployed to public May 14, 2026 03:01 Inactive

copy-pr-bot Bot temporarily deployed to test May 14, 2026 03:02 Inactive

copy-pr-bot Bot temporarily deployed to public May 14, 2026 03:05 Inactive

copy-pr-bot Bot temporarily deployed to public May 21, 2026 14:12 Inactive

chtruong814 and others added 2 commits May 21, 2026 22:24

Add speech CI build ID label

bcc4980

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

Fix SpeechLM2 automodel data parallel rank

f72d5ab

Signed-off-by: Piotr Żelasko <pzelasko@nvidia.com>

pzelasko added 2 commits May 27, 2026 07:56

Fix lint failures after dependency migration

88de033

Signed-off-by: Piotr Żelasko <pzelasko@nvidia.com>

Merge branch 'main' into chtruong/uv-lock

fa5e5f9

Signed-off-by: Piotr Żelasko <pzelasko@nvidia.com>

github-advanced-security AI found potential problems May 27, 2026

View reviewed changes

Comment thread nemo/collections/asr/data/feature_to_label.py Dismissed

Fix ASR CUDA graph transcribe workers

074a90a

artbataev reviewed May 27, 2026

View reviewed changes

Comment thread pyproject.toml Outdated

pzelasko added 2 commits May 27, 2026 14:29

Fix functional model support tests

5034b01

Use CUDA 13.2 PyTorch wheels

eca699c

Fix ASR CUDA graph fallback errors

1eca063

artbataev reviewed May 28, 2026

View reviewed changes

Comment thread nemo/collections/asr/parts/submodules/transducer_decoding/label_looping_base.py Outdated

This was referenced May 28, 2026

Remove unnecessary dependencies, remove some more unused modules #15494

Closed

Remove stale NeMo dependencies on uv base #15734

Draft

pzelasko approved these changes May 28, 2026

View reviewed changes

artbataev approved these changes May 28, 2026

View reviewed changes

Merge branch 'main' into chtruong/uv-lock

a4a9c66

thomasdhc approved these changes May 29, 2026

View reviewed changes

Conversation

chtruong814 commented May 13, 2026

Summary

Testing

Uh oh!

chtruong814 commented May 13, 2026

Uh oh!

chtruong814 commented May 14, 2026

Uh oh!

copy-pr-bot Bot commented May 27, 2026

Uh oh!

pzelasko commented May 27, 2026

Uh oh!

copy-pr-bot Bot commented May 27, 2026

Uh oh!

pzelasko commented May 27, 2026

Uh oh!

Uh oh!

pzelasko commented May 27, 2026

Uh oh!

Uh oh!

pzelasko commented May 27, 2026

Uh oh!

pzelasko commented May 28, 2026

Uh oh!

Uh oh!

pzelasko commented May 28, 2026

Uh oh!

github-actions Bot commented May 28, 2026

Uh oh!

pzelasko left a comment

Choose a reason for hiding this comment

Uh oh!

artbataev left a comment

Choose a reason for hiding this comment

Uh oh!

chtruong814 commented May 29, 2026

Uh oh!

github-actions Bot commented May 29, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

5 participants