nvidia

Documentation

Index

• type KindCluster
  • func NewKindCluster(env config.Env, vm *remote.Host, name string, clusterOpts *KindClusterOptions, ...) (*KindCluster, error)
• type KindClusterOption
  • func WithCudaSanityCheckImage(image string) KindClusterOption
  • func WithGPUOperatorVersion(version string) KindClusterOption
  • func WithHostGoVersion(version string) KindClusterOption
  • func WithKindImage(image string) KindClusterOption
  • func WithKubeVersion(version string) KindClusterOption
  • func WithNvkindVersion(version string) KindClusterOption
• type KindClusterOptions
  • func NewKindClusterOptions(opts ...KindClusterOption) *KindClusterOptions

Types

type KindCluster

type KindCluster struct {
	*kubernetes.Cluster

	GPUOperator *helm.Release
}

KindCluster represents a Kubernetes cluster that is GPU-enabled using nvkind. Both the cluster and the GPU operator should be depended on for resources that require GPU support.

func NewKindCluster

func NewKindCluster(env config.Env, vm *remote.Host, name string, clusterOpts *KindClusterOptions, opts ...pulumi.ResourceOption) (*KindCluster, error)

NewKindCluster creates a new Kubernetes cluster using nvkind so that nodes can be GPU-enabled. We need a different function rather than re-using kubernetes.NewKindCluster because we GPU-enabled kind clusters require a set of patches that aren't trivial. Instead of writing them all down here, we have decided to use the nvkind tool to create the cluster. This means that we cannot follow the same code path as for regular kind clusters.

type KindClusterOption

type KindClusterOption func(*KindClusterOptions)

KindClusterOption is a function that modifies a KindClusterOptions

func WithCudaSanityCheckImage

func WithCudaSanityCheckImage(image string) KindClusterOption

WithCudaSanityCheckImage sets the image to use for the CUDA sanity check commands. Note that the CUDA version in this image must be equal or less than the one installed in the AMI.

func WithGPUOperatorVersion

func WithGPUOperatorVersion(version string) KindClusterOption

WithGPUOperatorVersion sets the version of the Nvidia GPU operator to install

func WithHostGoVersion

func WithHostGoVersion(version string) KindClusterOption

WithHostGoVersion sets the version of Go to install in the host

func WithKindImage

func WithKindImage(image string) KindClusterOption

WithKindImage sets the image to use for the kind nodes. The version used by this image will be the one defined by kubernetes.GetKindVersionConfig based on the kubernetes version used.

func WithKubeVersion

func WithKubeVersion(version string) KindClusterOption

WithGPUOperatorVersion sets the version of the Nvidia GPU operator to install

func WithNvkindVersion

func WithNvkindVersion(version string) KindClusterOption

WithNvkindVersion sets the version of the nvkind utility to install

type KindClusterOptions

type KindClusterOptions struct {
	// contains filtered or unexported fields
}

KindClusterOptions contains the options for creating a kind cluster with the Nvidia GPU operator

func NewKindClusterOptions

func NewKindClusterOptions(opts ...KindClusterOption) *KindClusterOptions

NewKindClusterOptions creates a new KindClusterOptions with the given options, or defaults