Gebruik agentvaardigheden met AI Runtime

De databricks-ai-runtime agentvaardigheid geeft AI-codeeragenten, zoals Claude Code, Cursor en Codex, instructies voor het werken met AI Runtime via de databricks air Databricks CLI-commando's. Met de vaardigheid geïnstalleerd kun je je agent vragen een workload-YAML te schrijven, een GPU-trainingsrun in te dienen en de voortgang te controleren.

Requirements

De vaardigheid installeren

Voer de volgende opdracht uit:

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

Volg de instructies om je AI-agent en installatiescope te kiezen. Voor installatieopties, zie aitools commandogroep.

Wat de vaardigheid dekt

Capability Wat de agent kan doen
Workloads schrijven Zet een verzoek om in een workload-YAML met compute, omgeving, codebron en het launchcommando voor je framework, zoals torchrun, Accelerate of DeepSpeed.
Configureer runs Stel omgevingsvariabelen, geheimen, hyperparameters, herpogingen en permissies in.
Runs indienen Valideer de configuratie met --dry-run, dien deze vervolgens in met databricks air run en een idempotentiesleutel.
Controleer en monitor runs Haal status met databricks air get en databricks air list, stream logs met databricks air logs, en verwijs je naar de MLflow-run voor metrics.
Runs annuleren Stop één, meerdere of al je actieve runs met databricks air cancel.
Debugmislukkingen Haal logs uit een mislukte run en toon de fout.
Aangepaste Docker-installatiekopieën gebruiken Bouw en push een custom image, en verwijs die dan in de workload.

Aanvullende bronnen