gpu

package
v0.0.0-...-cd5b632 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Apr 25, 2020 License: Apache-2.0 Imports: 5 Imported by: 0

Documentation

Index

Constants

View Source
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"
)

Variables

This section is empty.

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

Types

This section is empty.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL