Skip to main content

Device

Represents a device type, its quantity and partition if applicable.

Attributes​

  • quantity: int

    • The number of devices of this type.

  • device: str | None

    • The type of device (e.g., "T4", "A100").

  • partition: str | None

    • The partition of the device (e.g., "1g.5gb", "2g.10gb" for gpus) or ("1x1", ... for tpus).