SIGN IN SIGN UP

ci: build wheel_base on the ubi8 base (public manylinux + NG… (#1991)

…C CUDA)

Incorporates the "ubi8" base image discussed on #1968: the wheel_base
deps stage now builds on the public PyPA manylinux_2_28 image and COPYs
CUDA from an el8/ubi8 NGC CUDA devel image
(nvcr.io/nvidia/cuda:*-devel-ubi8), instead of a combined internal
manylinux+CUDA base.

- Dockerfile.manylinux: add the Option-B header (cuda donor stage + quay
manylinux wheel_base + COPY --from=cuda + gcc-toolset-14 + CMake 4.x
policy). Keeps #1968's two-stage wheel_base/wheel caching split and the
opt-in --build-infinia / --build-ucx-spcx-plugin blocks untouched.
INFINIA stays opt-in (host-pull COPY) — not the bundled-default FROM
stage.
- build-container.sh: restore --cuda-version (MAJOR.MINOR) so
local/release full builds drive the torch cuXXX index + cu12/cu13
meta-wheel split.
- build-wheel-matrix.yaml: point BASE_IMAGE/BASE_TAG at the ubi8 CUDA
image, add CUDA_VERSION + MANYLINUX_VERSION, pass --cuda-version, and
forward CUDA_VERSION/MANYLINUX_VERSION to the wheel_base cache build.
- Bump CI_IMAGE_TAG in all six matrix YAMLs (cidemo-init.sh requires it
when a CI file such as Dockerfile.manylinux changes).

Note: only the ci-demo wheel_base cache build pulls nvcr.io/quay.io;
per-PR wheel builds pull the cached wheel_base from Artifactory. If the
blossom runner cannot reach nvcr.io/quay.io, repoint BASE_IMAGE at an
Artifactory ubi8 mirror (same ubi8 tag).

## What?
_Describe what this PR is doing._

## Why?
_Justification for the PR. If there is an existing issue/bug, please
reference it. For
bug fixes, the 'Why?' and 'What?' can be merged into a single item._

## How?
_It is optional, but for complex PRs, please provide information about
the design,
architecture, approach, etc._


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

* **Chores**
* Updated wheel builds to use pinned CUDA and manylinux versions for
more consistent, repeatable packaging.
* Added a selectable CUDA version option for containerized wheel builds
(defaulting to CUDA 13.0).
* Improved build compatibility by aligning the toolchain/CMake behavior
with newer environments.
* Refreshed the wheel build base image approach while preserving CUDA
support for the build.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->

---------

Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
N
Nate Mailhot committed
a9a66cd31325c1803fc6a367416e111b77b4d9e3
Parent: 7a244b9
Committed by GitHub <noreply@github.com> on 7/29/2026, 2:36:31 AM