Používejte dovednosti agenta s AI Runtime

Dovednostdatabricks-ai-runtime agenta dává AI kódovacím agentům, jako jsou Claude Code, Cursor a Codex, instrukce pro práci s AI Runtime prostřednictvím příkazů databricks air Databricks CLI. S touto dovedností můžete požádat agenta, aby napsal workload YAML, odeslal tréninkový běh na GPU a zkontroloval jeho průběh.

Requirements

Nainstalujte dovednost

Spusťte následující příkaz:

databricks aitools install --skills-only --skills databricks-ai-runtime

Postupujte podle pokynů k výběru AI agenta a rozsahu instalace. Pro možnosti instalace viz aitools skupinu příkazů.

Co dovednost pokrývá

Schopnost Co agent může dělat
Zapisovací úlohy Přeměňte požadavek na workload YAML s výpočtem, prostředím, kódovým zdrojem a launch příkazem pro váš framework, například torchrun, Accelerate nebo DeepSpeed.
Konfigurace běhů Nastavte proměnné prostředí, tajné klíče, hyperparametry, opakované pokusy a oprávnění.
Odeslat běhy Ověřte konfiguraci pomocí --dry-run, pak odešlete pomocí databricks air run a klíče idempotence.
Kontrolovat a sledovat běhy Získejte stav pomocí databricks air get a databricks air list, streamujte logy pomocí databricks air logs a odkáže vás na běh MLflow s metrikami.
Zrušit spuštění Zrušte jedno, několik nebo všechna svá aktivní spuštění pomocí databricks air cancel.
Chyby při ladění Vytáhněte logy z neúspěšného běhu a zobrazte chybu.
Použití vlastních imagí Dockeru Postavte a nahrajte vlastní image, pak na něj ve workloadu odkažte.

Dodatečné zdroje