d:

zenml step-operator flavor list

How to use itYou don't need to directly interact with any ZenML step operator in your code. As long as the step operator that you want to use is part of your active ZenML stack, you can simply specify it in the @step decorator of your step.

from zenml import step

@step(step_operator= <STEP_OPERATOR_NAME>)

def my_step(...) -> ...:

...

Specifying per-step resources

If your steps require additional hardware resources, you can specify them on your steps as described here.

Enabling CUDA for GPU-backed hardware

Note that if you wish to use step operators to run steps on a GPU, you will need to follow the instructions on this page to ensure that it works. It requires adding some extra settings customization and is essential to enable CUDA for the GPU to give its full acceleration.

PreviousDevelop a Custom Model Deployer

NextAmazon SageMaker

Last updated 15 days ago