gpu

Documentation

Index

• Constants
• func GetGPUDevicePluginImage() string
• func NVIDIADevicePlugin() *v1.Pod
• func NumberOfNVIDIAGPUs(node *v1.Node) int64

Constants

const (
	// NVIDIAGPUResourceName is the extended name of the GPU resource since v1.8
	// this uses the device plugin mechanism
	NVIDIAGPUResourceName = "nvidia.com/gpu"

	// GPUDevicePluginDSYAML is the official Google Device Plugin Daemonset NVIDIA GPU manifest for GKE
	// TODO: Parametrize it by making it a feature in TestFramework.
	// so we can override the daemonset in other setups (non COS).
	GPUDevicePluginDSYAML = "https://raw.githubusercontent.com/kubernetes/kubernetes/master/cluster/addons/device-plugins/nvidia-gpu/daemonset.yaml"
)

Functions

func GetGPUDevicePluginImage

func GetGPUDevicePluginImage() string

GetGPUDevicePluginImage returns the image of GPU device plugin.

func NVIDIADevicePlugin

func NVIDIADevicePlugin() *v1.Pod

NVIDIADevicePlugin returns the official Google Device Plugin pod for NVIDIA GPU in GKE

func NumberOfNVIDIAGPUs

func NumberOfNVIDIAGPUs(node *v1.Node) int64

NumberOfNVIDIAGPUs returns the number of GPUs advertised by a node This is based on the Device Plugin system and expected to run on a COS based node After the NVIDIA drivers were installed TODO make this generic and not linked to COS only