package gpu

import "k8s.io/kubernetes/test/e2e/framework/gpu"

Index

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 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

Source Files

gpu_util.go

Version
v1.17.17
Published
Jan 13, 2021
Platform
js/wasm
Imports
4 packages
Last checked
1 minute ago –

Tools for package owners.