common

Documentation

Index

• func GPUNeedsFabricManager(size string) bool
• func GetAKSGPUImageSHA(size string) string
• func GetCommaSeparatedGPUSizes() string
• func GetCommaSeparatedMarinerGPUSizes() string
• func GetGPUDriverVersion(size string) string
• func IsMIGNode(gpuInstanceProfile string) bool
• func IsMarinerEnabledGPUSKU(vmSize string) bool
• func IsNvidiaEnabledSKU(vmSize string) bool
• func UseWindowsCudaGPUDriver(vmSize string) bool
• func UseWindowsGridGPUDriver(vmSize string) bool

Functions

func GPUNeedsFabricManager

func GPUNeedsFabricManager(size string) bool

func GetAKSGPUImageSHA

func GetAKSGPUImageSHA(size string) string

func GetCommaSeparatedGPUSizes

func GetCommaSeparatedGPUSizes() string

func GetCommaSeparatedMarinerGPUSizes

func GetCommaSeparatedMarinerGPUSizes() string

func GetGPUDriverVersion

func GetGPUDriverVersion(size string) string

NV series GPUs target graphics workloads vs NC which targets compute. they typically use GRID, not CUDA drivers, and will fail to install CUDA drivers. NVv1 seems to run with CUDA, NVv5 requires GRID. NVv3 is untested on AKS, NVv4 is AMD so n/a, and NVv2 no longer seems to exist (?).

func IsMIGNode

func IsMIGNode(gpuInstanceProfile string) bool

IsMIGNode check if the node should be partitioned.

func IsMarinerEnabledGPUSKU

func IsMarinerEnabledGPUSKU(vmSize string) bool

IsMarinerNvidiaEnabledSKU determines if an Mariner VM SKU has nvidia driver support.

func IsNvidiaEnabledSKU

func IsNvidiaEnabledSKU(vmSize string) bool

IsNvidiaEnabledSKU determines if an VM SKU has nvidia driver support.

func UseWindowsCudaGPUDriver

func UseWindowsCudaGPUDriver(vmSize string) bool

func UseWindowsGridGPUDriver

func UseWindowsGridGPUDriver(vmSize string) bool