Environment Variables

Environment variables for serverless GPU configuration. All environment variables are optional and must be set outside of the @distributed call.

Pool Selection

DATABRICKS_USE_RESERVED_GPU_POOL

Whether to launch remote workloads on a reserved GPU pool. When set to a falsy value, on-demand compute is used instead.

Default:

true

os.environ["DATABRICKS_USE_RESERVED_GPU_POOL"] = "false"

DATABRICKS_REMOTE_GPU_POOL_ID

ID of a specific reserved GPU pool to launch on. When unset, the API auto-detects the notebook session’s attached pool. When set, it overrides the auto-detected pool. To list available pools, run sgcli get pools.

Default:

unset

os.environ["DATABRICKS_REMOTE_GPU_POOL_ID"] = "your-pool-id"

Logging

DATABRICKS_SERVERLESS_GPU_LOG_VERBOSE

Whether to enable verbose logging for the Serverless GPU API.

Default:

false

os.environ["DATABRICKS_SERVERLESS_GPU_LOG_VERBOSE"] = "true"