Använd agentfärdigheter med AI Runtime

agentkompetensendatabricks-ai-runtime ger AI-kodningsagenter, såsom Claude Code, Cursor och Codex, instruktioner för att arbeta med AI Runtime via databricks air Databricks CLI-kommandon. Med skill installerad kan du be din agent att skriva en workload-YAML, skicka in en GPU-träningskörning och kontrollera dess framsteg.

Requirements

Installera färdigheten

Kör följande kommando:

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

Följ instruktionerna för att välja din AI-agent och installationsomfång. För installationsalternativ, se aitools kommandogruppen.

Vad färdigheten täcker

Capability Vad agenten kan göra
Skrivarbetslaster Gör om en förfrågan till en workload-YAML med beräkningsresurser, miljö, kodkälla och startkommando för ditt ramverk, såsom torchrun, Accelerate eller DeepSpeed.
Konfigurera körjobb Ställ in miljövariabler, hemliga värden, hyperparametrar, återförsök och behörigheter.
Skicka in körningar Verifiera konfigurationen med --dry-run, skicka sedan in med databricks air run och en idempotensnyckel.
Kontrollera och övervaka körningar Få status med databricks air get och databricks air list, strömma loggar med databricks air logs, och visa dig MLflow-körningen för mätvärden.
Avbryt körningar Avbryt en, flera eller alla dina aktiva körningar med databricks air cancel.
Debuggningsfel Ta ut loggar från en misslyckad körning och visa felet.
Använda anpassade Docker-avbildningar Bygg och pusha en anpassad avbild och referera sedan till avbilden i arbetslasten.

Ytterligare resurser