Instalación de la CLI de AI Runtime

Important

La CLI del entorno de ejecución de IA está en beta.

Instale la air CLI con uv y autentíquela con un área de trabajo de Databricks usando un perfil de Databricks CLI. La CLI requiere Python 3.10 o posterior.

Requirements

Instalación de la CLI

Databricks recomienda instalar la CLI con uv:

uv tool install --force databricks-air --python 3.12

uv tool install coloca air en su propio entorno aislado y lo expone en su PATH, por lo que no entra en conflicto con el intérprete de Python que usa para el código de entrenamiento.

--python 3.12 se recomienda, pero opcional. Si no especifica la versión Python, uv usa la versión disponible más reciente que satisface la restricción Python del paquete.

Si aún no tiene uv, instálelo primero:

curl -LsSf https://astral.sh/uv/install.sh | sh

Compruebe la instalación:

air --version
air --help

Autenticación

La CLI de AI Runtime reutiliza los perfiles de autenticación de la CLI de Databricks. Inicie sesión en el área de trabajo y asigne un nombre al perfil cuando se le solicite:

databricks auth login --host https://<your-workspace>.cloud.databricks.com

Pase el nombre del perfil a cualquier comando air con -p. Por ejemplo:

air list runs -p my-workspace

Como alternativa, establezca DATABRICKS_CONFIG_PROFILE en el intérprete de comandos para establecer un perfil como predeterminado:

export DATABRICKS_CONFIG_PROFILE=my-workspace

Para ver todas las opciones de autenticación, consulte Autenticación para la CLI de Databricks.

Recursos adicionales

Después de instalar, defina las cargas de trabajo en un archivo train.yaml config con dependencias en línea. Comience con el inicio rápido y, a continuación, use la referencia de YAML a medida que compile la configuración: