feat: azurelinux add nvidia vgpu driver installation selection#7986
Merged
feat: azurelinux add nvidia vgpu driver installation selection#7986
Conversation
miz060
force-pushed
the
mitchzhu/azl-grid-gpu_driver-pr
branch
from
8895c0f to
f869d83
Compare
Contributor
There was a problem hiding this comment.
Pull request overview
Adds NVIDIA GRID (vGPU guest) driver selection for converged A10 SKUs on Azure Linux/Mariner by branching GPU driver installation based on NVIDIA_GPU_DRIVER_TYPE, and extends ShellSpec coverage for the routing logic.
Changes:
- Add
downloadGridDrivers()and route converged SKUs (NVIDIA_GPU_DRIVER_TYPE=grid) to GRID installation indownloadGPUDrivers(). - Add ShellSpec tests validating GRID vs CUDA vs CUDA-open routing behavior.
Reviewed changes
Copilot reviewed 2 out of 2 changed files in this pull request and generated 6 comments.
| File | Description |
|---|---|
parts/linux/cloud-init/artifacts/mariner/cse_install_mariner.sh |
Introduces GRID driver install function and selects GRID vs CUDA driver flow based on NVIDIA_GPU_DRIVER_TYPE. |
spec/parts/linux/cloud-init/artifacts/cse_install_mariner_spec.sh |
Adds tests that validate the new selection/routing behavior without performing real downloads. |
parts/linux/cloud-init/artifacts/mariner/cse_install_mariner.sh
Outdated
Show resolved
Hide resolved
parts/linux/cloud-init/artifacts/mariner/cse_install_mariner.sh
Outdated
Show resolved
Hide resolved
parts/linux/cloud-init/artifacts/mariner/cse_install_mariner.sh
Outdated
Show resolved
Hide resolved
parts/linux/cloud-init/artifacts/mariner/cse_install_mariner.sh
Outdated
Show resolved
Hide resolved
Add GRID(vGPU) driver installation logic for azurelinux with A10 GPU VM sizes Signed-off-by: Mitch Zhu <mitchzhu@microsoft.com>
miz060
force-pushed
the
mitchzhu/azl-grid-gpu_driver-pr
branch
from
f869d83 to
9730e58
Compare
miz060
requested review from
AbelHu,
Devinwong,
YaoC,
awesomenix,
cameronmeissner,
djsly,
ganeshkumarashok,
juan-lee,
junjiezhang1997,
lilypan26,
mxj220,
phealy,
r2k1,
timmy-wright,
yewmsft and
zachary-bailey
as code owners
miz060
requested review from
calvin197,
pdamianov-dev,
sulixu and
surajssd
as code owners
awesomenix
approved these changes
Mar 10, 2026
lilypan26
reviewed
Mar 10, 2026
| # Converged GPU sizes (NVads_A10_v5, NCads_A10_v4) require NVIDIA GRID (vGPU guest) | ||
| # drivers instead of CUDA drivers. | ||
| GRID_PACKAGE=$(dnf repoquery -y --available "nvidia-vgpu-guest-driver*" | \ | ||
| grep -E "nvidia-vgpu-guest-driver-[0-9]+.*_${KERNEL_VERSION}" | sort -V | tail -n 1) |
Contributor
There was a problem hiding this comment.
just checked on a node and i think the package has the architecture at the end like so: nvidia-vgpu-guest-driver-0:570.195.03-1_6.6.126.1.1.azl3.x86_64. So i thnk you need another * asterisk at the end.
Member
Author
There was a problem hiding this comment.
I see.
Through AB-E2E I found that
grep regex: nvidia-vgpu-guest-driver-[0-9]+.*_${KERNEL_VERSION} is not anchored with $, so it matches the kernel version substring regardless of what comes after it (.x86_64). This will work with or without the extra asterisk.
lilypan26
approved these changes
Mar 11, 2026
janenotjung-hue
pushed a commit
that referenced
this pull request
Mar 11, 2026
Signed-off-by: Mitch Zhu <mitchzhu@microsoft.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
What this PR does / why we need it:
Converged GPU sizes (NVads_A10_v5, NCads_A10_v4) require NVIDIA GRID vGPU guest drivers instead of standard CUDA drivers. Previously, AzureLinux 3.0 had no GRID driver support and cannot support these sizes. This PR adds azurelinux GRID driver installation logic, routing converged sizes to the GRID driver path based on NVIDIA_GPU_DRIVER_TYPE while leaving the existing cuda/cuda-open selection unchanged for all other GPU SKUs.
Which issue(s) this PR fixes:
Fixes #
Validation: