GPU

Create a GPU device instance.

def GPU(
    device: GPUType,
    quantity: GPUQuantity,
    partition: A100Parts | A100_80GBParts | H100Parts | H200Parts | None = None
) - > Device

Create a GPU device instance.

Parameters

Name	Type	Description
device	GPUType	The type of GPU (e.g., "T4", "A100").
quantity	GPUQuantity	The number of GPUs of this type.
partition	`A100Parts	A100_80GBParts

Returns

Type	Description
Device	Device instance.