Skip to content

modify ldconfig in ubi image so nvidia libraries are on path - #1644

Merged
rapids-bot[bot] merged 1 commit into
NVIDIA:mainfrom
tmckayus:ubipatch
Jul 31, 2026
Merged

modify ldconfig in ubi image so nvidia libraries are on path#1644
rapids-bot[bot] merged 1 commit into
NVIDIA:mainfrom
tmckayus:ubipatch

Conversation

@tmckayus

Copy link
Copy Markdown
Contributor

under rhel, libcuopt and other libs end up under different paths so relative references don't work. This causes the cuopt_grpc_server to fail in the ubi10 image

$ docker run --rm --gpus=all --network=host -e CUOPT_SERVER_TYPE=grpc nvidia/cuopt:26.8.0a-cu13-ubi10
/usr/local/lib64/python3.14/site-packages/libcuopt/bin/cuopt_grpc_server: error while loading shared libraries: libnccl.so.2: cannot open shared object file: No such file or directory

with fix:

$ docker run --rm --gpus=all --network=host -e CUOPT_SERVER_TYPE=grpc cuopt-ubi10-nccl-fixtest:latest
[2026-07-31 00:01:33.702] cuOpt gRPC Remote Solve Server
[2026-07-31 00:01:33.702] ==============================
[2026-07-31 00:01:33.702] Port: 5001
[2026-07-31 00:01:33.702] Workers: 1
[2026-07-31 00:01:33.730] [Server] 1 GPU(s) available for worker assignment
[2026-07-31 00:01:33.734] [Worker 0] Started (PID: 58)
[2026-07-31 00:01:33.735] [Server] Result retrieval thread started
[2026-07-31 00:01:33.735] [Server] Incumbent retrieval thread started
[2026-07-31 00:01:33.735] [Server] Worker monitor thread started
[2026-07-31 00:01:33.736] [Server] Session reaper thread started (timeout=300s)
[2026-07-31 00:01:33.742] [gRPC Server] Listening on 0.0.0.0:5001
[2026-07-31 00:01:33.742] [gRPC Server] Workers: 1
[2026-07-31 00:01:33.742] [gRPC Server] Max message size: 268435456 bytes (256 MiB)
[2026-07-31 00:01:33.742] [gRPC Server] Press Ctrl+C to shutdown
[2026-07-31 00:01:33.891] [Worker] RMM pool size: 1 GiB
[2026-07-31 00:01:33.891] [Worker 0] Using CUDA device 0 of 1

@tmckayus
tmckayus requested a review from a team as a code owner July 31, 2026 00:03
@tmckayus tmckayus added bug Something isn't working non-breaking Introduces a non-breaking change P0 labels Jul 31, 2026
@tmckayus tmckayus added this to the 26.08 milestone Jul 31, 2026
@github-actions

Copy link
Copy Markdown

CI Test Summary

⏭️ All 5 test job(s) skipped.

@coderabbitai

coderabbitai Bot commented Jul 31, 2026

Copy link
Copy Markdown

Review Change Stack

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info
⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: ed01c2ef-d049-422f-ac5d-5110c473fa37

📥 Commits

Reviewing files that changed from the base of the PR and between 984ce71 and 1625b7a.

📒 Files selected for processing (1)
  • ci/docker/Dockerfile.ubi
🚧 Files skipped from review as they are similar to previous changes (1)
  • ci/docker/Dockerfile.ubi

📝 Walkthrough

Walkthrough

Changes

NVIDIA linker configuration

Layer / File(s) Summary
Configure and validate NVIDIA libraries
ci/docker/Dockerfile.ubi
The UBI Docker build discovers NVIDIA wheel libraries in lib and lib64 site-packages paths, updates the dynamic linker configuration, runs ldconfig, and verifies libnccl.so.2 is registered.

Estimated code review effort: 2 (Simple) | ~10 minutes

Suggested reviewers: msarahan

🚥 Pre-merge checks | ✅ 5
✅ Passed checks (5 passed)
Check name Status Explanation
Title check ✅ Passed The title clearly identifies the main change: updating ldconfig in the UBI image to expose NVIDIA libraries.
Description check ✅ Passed The description explains the RHEL library path issue, the missing libnccl.so.2 error, and the successful runtime test.
Docstring Coverage ✅ Passed No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check ✅ Passed Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check ✅ Passed Check skipped because no linked issues were found for this pull request.
✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests

Comment @coderabbitai help to get the list of available commands.

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@ci/docker/Dockerfile.ubi`:
- Around line 90-93: Update the NCCL validation in the Dockerfile command chain
after ldconfig to match libnccl.so.2 entries specifically under the discovered
/nvidia/nccl/lib directory, while allowing versioned filenames rather than
requiring an exact unversioned name. Keep the existing wheel-directory
configuration and ldconfig steps unchanged.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: 45060e5f-630d-4072-bcdf-0bd19db370e4

📥 Commits

Reviewing files that changed from the base of the PR and between e72fcbc and 984ce71.

📒 Files selected for processing (1)
  • ci/docker/Dockerfile.ubi

Comment thread ci/docker/Dockerfile.ubi

@ramakrishnap-nv ramakrishnap-nv left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Rest looks good, and if possible add a test run this in ubi image.

under rhel, libcuopt and other libs end up under different paths
so relative references don't work.
@ramakrishnap-nv

Copy link
Copy Markdown
Collaborator

/merge

@rapids-bot
rapids-bot Bot merged commit ef0d785 into NVIDIA:main Jul 31, 2026
23 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working non-breaking Introduces a non-breaking change P0

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants