Azure / Azure/AKS

Update GPU driver type for Windows GPU

Open
#4,505 3 comments 0 reactions 1 assignee Assigned to @sdesai345 View on GitHub
feature-request GPU windows
Dominant language
TypeScript
Stars
2.1k
Forks
395
Avg merge
2d 22h
Merged PRs (30d)
13

Description

### GA ETA: No current ETA
Note: This is an estimate and is subject to change

---
Currently, a specific GPU Driver for each VM size is installed automatically. Because workload and driver compatibility are important for functioning GPU workloads, AKS will introduce a new property which will allow you to customize the driver type for your GPU node. This feature is not supported for Linux GPU node pools.

When creating a Windows agent pool in Azure Kubernetes Service (AKS) with GPU support, you have the option to explicitly specify the type of GPU driver using the driverType property within the GPUProfile. The available options are:

- GRID: For applications requiring virtualization support.
- CUDA: Optimized for computational tasks in scientific computing and data-intensive applications.

Note: When you explicitly set the driverType property, you assume responsibility for ensuring that the selected driver type is compatible with the specific VM size and configuration of your node pool. While AKS attempts to validate compatibility, there are scenarios where the node pool creation might fail due to incompatibilities between the specified driver type and the underlying VM or hardware.

**This feature is available in preview.**

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.