flytekit.Resources#

class flytekit.Resources(cpu=None, mem=None, gpu=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]) –
ephemeral_storage (Optional[str]) –

Return type

None

cpu: Optional[str] = None

ephemeral_storage: Optional[str] = None

gpu: Optional[str] = None

mem: Optional[str] = None