Instalar a CLI do AI Runtime

Importante

A CLI do AI Runtime está em Beta.

Instale a CLI air com uv e autentique-a em um espaço de trabalho do Databricks usando um perfil da Databricks CLI. A CLI requer Python 3.10 ou mais recente.

Requirements

  • Python 3.10 ou mais recente.
  • Um workspace do Databricks com o AI Runtime habilitado. Confira os Requisitos
  • O Databricks CLI, que gerencia perfis de autenticação em ~/.databrickscfg.

Instalar a CLI

O Databricks recomenda a instalação da CLI com uv:

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

uv tool install coloca air em seu próprio ambiente isolado e o expõe em seu PATH, para que ele não entre em conflito com o interpretador de Python que você usa para seu código de treinamento.

--python 3.12 é recomendado, mas opcional. Se você não especificar a versão Python, uv usará a versão mais recente disponível que satisfaça a restrição de Python do pacote.

Se você ainda não tiver uv, instale-o primeiro:

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

Verifique a instalação:

air --version
air --help

Authenticate

A CLI do AI Runtime reutiliza perfis de autenticação da CLI do Databricks. Faça logon no workspace e nomeie o perfil quando solicitado:

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

Passe o nome do perfil para qualquer comando air com -p. Por exemplo:

air list runs -p my-workspace

Como alternativa, defina DATABRICKS_CONFIG_PROFILE em seu shell para tornar um perfil o padrão:

export DATABRICKS_CONFIG_PROFILE=my-workspace

Para todas as opções de autenticação, consulte Autenticação para a CLI do Databricks.

Recursos adicionais

Após a instalação, defina cargas de trabalho em uma train.yaml configuração com dependências embutidas. Comece com o início rápido e use a referência YAML ao compilar sua configuração:

  • Last updated on