Wykorzystaj umiejętności agenta z AI Runtime

funkcja agent skilldatabricks-ai-runtime przekazuje agentom kodującym AI, takim jak Claude Code, Cursor i Codex, instrukcje dotyczące pracy z AI Runtime za pomocą databricks air poleceń Databricks CLI. Po zainstalowaniu umiejętności możesz poprosić agenta o napisanie YAML obciążenia, przesłanie zadania treningowego GPU i sprawdzenie postępów.

Requirements

Zainstaluj umiejętność

Uruchom następujące polecenie:

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

Podążaj za instrukcjami, aby wybrać agenta AI i zakres instalacji. Opcje instalacji znajdziesz w grupieaitools poleceń.

Co obejmuje ta umiejętność

Capability Co może zrobić agent
Obciążenia zapisu Przekształć żądanie w workload YAML z obliczeniami, środowiskiem, kodem źródłowym i poleceniem startowym dla swojego frameworka, takim jak torchrun, Accelerate lub DeepSpeed.
Konfiguruj uruchomienia Ustaw zmienne środowiskowe, sekrety, hiperparametry, ponowienia i uprawnienia.
Prześlij uruchomienia Zweryfikuj konfigurację za pomocą --dry-run, a następnie wyślij za pomocą databricks air run i klucza idempotencji.
Sprawdzaj i monitoruj uruchomienia Uzyskaj status za pomocą databricks air get i databricks air list, przesyłaj strumieniowo logi za pomocą databricks air logs, i przekieruje Cię do uruchomienia MLflow dla metryk.
Anuluj uruchomienia Anuluj jedno, kilka lub wszystkie aktywne uruchomienia za pomocą databricks air cancel.
Błędy debugowania Pobierz logi z nieudanego uruchomienia i wyświetl błąd.
Użyj niestandardowych obrazów Dockera Zbuduj i wypchnij własny obraz kontenera, a następnie odwołaj się do niego w obciążeniu roboczym.

Dodatkowe zasoby