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
Return type

None

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