Use a CLI do Databricks com AI Runtime

info

Visualização

Os comandos do Databricks CLI para o AI Runtime estão em Visualização pública. Um administrador do Workspace deve ativar a versão preliminar do AI Runtime antes que você possa enviar cargas de trabalho. Consulte Requisitos.

O grupo de comandos databricks air envia e gerencia cargas de trabalho de treinamento distribuído no AI Runtime, a plataforma de compute de GPU serverless sob demanda. Ele usa configuração de job baseada em YAML, integra-se ao MLflow e oferece suporte a fluxos de trabalho de código baseados em workspace e em git.

The Databricks CLI commands for AI Runtime meet the same compliance standards as AI Runtime workloads. Para os padrões suportados e exceções, consulte compliance security profile.

Migrar da CLI do Python

A CLI do Databricks substitui a CLI air baseada em Python distribuída como o pacote databricks-air. A CLI do Python não recebe novos recursos. Instale a versão mais recente da CLI do Databricks para usar os recursos mais recentes do AI Runtime. A CLI do Databricks é compatível com macOS, Linux e Windows.

Alguns nomes de comandos e sinalizadores diferem, portanto, a migração exige mais do que apenas adicionar o prefixo databricks. Consulte diferenças de comando.

Quando usar a CLI

Use a CLI do Databricks com o AI Runtime quando quiser:

• Envie cargas de trabalho de treinamento de GPU de seu laptop e editor de código sem abrir um notebook.
• Defina jobs de treinamento de forma declarativa em YAML para que possam ser adicionados ao controle de código-fonte.
• Execução de Jobs de treinamento de longa duração ou treinamento distribuído em vários nós — workloads que precisam sobreviver a uma sessão interativa ou abranger mais de um nó.

Para a API Python no Notebook (@distributed e @ray_launch), consulte treinamento distribuído em Notebooks.

Para trabalhar interativamente em um único nó de GPU via SSH — a partir do seu terminal ou IDE, em vez de enviar uma workload — use databricks ssh connect. Consulte Conectar-se ao Databricks usando um túnel SSH.

Começar

Instale a CLI do Databricks mais recente e configure a autenticação. Consulte Install or update the Databricks CLI e Authentication for the Databricks CLI.

Em seguida, siga o início rápido para enviar sua primeira carga de trabalho. Para comandos e sinalizadores, consulte grupo de comandosair.