Adding a GPU to a function

Beta9 offers a fast way to spin up GPU runtimes for development.

You can add a GPU to your runtime by passing a gpu argument to your function decorator.

You must ensure your GPU type is registered as a provider. You can see a sample provider for EC2 here.

from beta9 import endpoint


@endpoint(gpu="A100-80")
def handler():
    print("📡 This is running on a GPU!")
    return {}