`nemo_microservices.types.jobs.gpu_execution_provider`#

Module Contents#

Classes#

GPUExecutionProvider

API#

class nemo_microservices.types.jobs.gpu_execution_provider.GPUExecutionProvider(/, **data: Any)#

Bases: nemo_microservices._models.BaseModel

container: nemo_microservices.types.jobs.container_spec.ContainerSpec#

None

Specification for a container configuration.

Defines the container image and related configuration for job execution.

num_gpus: int | None#: None

profile: str | None#: None

provider: typing_extensions.Literal[gpu] | None#: None

resources: nemo_microservices.types.jobs.compute_resources.ComputeResources | None#

None

Resource requirements matching k8s ResourceRequirements format.