Installation | NVIDIA Kumo

This page describes the Kumo Studio SDK. For local Python execution in an Azure Databricks notebook, follow Azure Databricks Notebook Setup; its package and identity procedure is different.

The Kumo Studio SDK supports Python 3.10 and later with a minimal set of dependencies. Installation steps are identical for the SaaS, Databricks, and Snowflake editions of Kumo Studio.

Before installing, ensure pip is available in your environment.

Install the SDK:

pip install kumo-studio

Verify the installation using its kumoai_studio Python import:

>>> import kumoai_studio as kumo
>>> kumo.__version__
'2.22.0'

The version number shown may differ from the example above.