flytekit.Resources

class flytekit.Resources(cpu=None, mem=None, gpu=None, storage=None, ephemeral_storage=None)[source]

This class is used to specify both resource requests and resource limits.

Resources(cpu="1", mem="2048")  # This is 1 CPU and 2 KB of memory
Resources(cpu="100m", mem="2Gi")  # This is 1/10th of a CPU and 2 gigabytes of memory

# For Kubernetes-based tasks, pods use ephemeral local storage for scratch space, caching, and for logs.
# This allocates 1Gi of such local storage.
Resources(ephemeral_storage="1Gi")

Note

Persistent storage is not currently supported on the Flyte backend.

Please see the User Guide for detailed examples. Also refer to the K8s conventions.

Methods

Attributes

Parameters

  • cpu (Optional[str]) –

  • mem (Optional[str]) –

  • gpu (Optional[str]) –

  • storage (Optional[str]) –

  • ephemeral_storage (Optional[str]) –

Return type

None

cpu: Optional[str] = None
ephemeral_storage: Optional[str] = None
gpu: Optional[str] = None
mem: Optional[str] = None
storage: Optional[str] = None