Skip to content

Conversation

@miz060
Copy link
Member

@miz060 miz060 commented Jan 23, 2026

What this PR does / why we need it:
AzureLinux 3.0 provides two CUDA driver packages: proprietary (cuda) for legacy GPUs and open-source (cuda-open) for newer GPUs. Previously, AzureLinux 3.0 always installed the proprietary driver. This PR adds GPU SKU detection to install cuda-open for A100/H100/H200/GB200 SKUs while continuing to use the proprietary cuda driver for legacy SKUs (T4, V100) that require it.

Which issue(s) this PR fixes:

Fixes #

Validation:

@miz060 miz060 changed the title feat: AzureLinux add Nvidia OpenRM driver installation selection feat: azurelinux add nvidia openrm driver installation selection Jan 23, 2026
@miz060 miz060 requested review from YaoC and yewmsft as code owners January 23, 2026 22:17
@miz060 miz060 changed the title feat: azurelinux add nvidia openrm driver installation selection feat: azurelinux add nvidia open driver installation selection Jan 23, 2026
return "cuda"
}

func UseOpenGPUDriver(size string) bool {
Copy link
Contributor

Choose a reason for hiding this comment

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

This requires changes in scriptless as well.

Copy link
Contributor

Choose a reason for hiding this comment

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

func getCSEEnv(config *aksnodeconfigv1.Configuration) map[string]string {

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants