Documentation
¶
Overview ¶
Package gpu implements GPU plugin.
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Config ¶
type Config struct {
Logger *zap.Logger
LogWriter io.Writer
Stopc chan struct{}
EKSConfig *eksconfig.Config
K8SClient k8s_client.EKS
}
Config defines GPU configuration.
type Tester ¶
type Tester interface {
// Name returns the name of the tester.
Name() string
// InstallNvidiaDriver installs the Nvidia device plugin for Kubernetes.
// After GPU worker nodes join the cluster, one must apply the Nvidia
// device plugin for Kubernetes as a DaemonSet.
// ref. https://docs.aws.amazon.com/eks/latest/userguide/create-managed-node-group.html
// ref. https://docs.aws.amazon.com/eks/latest/userguide/gpu-ami.html
// ref. https://github.com/NVIDIA/k8s-device-plugin
InstallNvidiaDriver() error
// CreateNvidiaSMI launches a pod manifest that launches a Cuda container that
// runs "nvidia-smi" on a GPU worker node.
// ref. https://docs.aws.amazon.com/eks/latest/userguide/gpu-ami.html
CreateNvidiaSMI() error
}
Tester defines GPU tester.
Click to show internal directories.
Click to hide internal directories.