Remarque
L’accès à cette page nécessite une autorisation. Vous pouvez essayer de vous connecter ou de modifier des répertoires.
L’accès à cette page nécessite une autorisation. Vous pouvez essayer de modifier des répertoires.
Le databricks-ai-runtime compétence d’agent fournit aux agents de codage IA, tels que Claude Code, Cursor et Codex, des instructions pour travailler avec AI Runtime via les databricks air commandes de la ligne de commande Databricks. Avec le skill installé, vous pouvez demander à votre agent d’écrire un YAML de charge de travail, de soumettre une session d’entraînement GPU et de vérifier sa progression.
Requirements
- La CLI Databricks, authentifiée auprès de votre espace de travail. Les exemples de la compétence utilisent le format aplati
code_source, qui nécessite la version 1.20.0 ou supérieure de Databricks CLI. - Un espace de travail avec le Runtime IA activé. Voir les exigences.
Installer la compétence
Exécutez la commande suivante:
databricks aitools install --skills-only --skills databricks-ai-runtime
Suivez les instructions pour choisir votre agent IA et le périmètre d’installation. Pour connaître les options d'installation, consultez le groupe de commandes aitools.
Ce que couvre la compétence
| Capacité | Ce que l’agent peut faire |
|---|---|
| Charges de travail d’écriture | Transformez une requête en YAML de charge de travail avec le calcul, l’environnement, la source de code et la commande launch de votre framework, comme torchrun, Accelerate ou DeepSpeed. |
| Configurer les exécutions | Définissez les variables d’environnement, les secrets, les hyperparamètres, les nouvelles tentatives et les permissions. |
| Envoyer les exécutions | Validez la configuration avec --dry-run, puis soumettez avec databricks air run et une clé d’idempotence. |
| Vérifier et surveiller les exécutions | Obtenez le statut avec databricks air get et databricks air list, consultez les journaux en continu avec databricks air logs, et accédez à l’exécution MLflow pour les métriques. |
| Annuler les exécutions | Annulez une, plusieurs ou toutes vos exécutions actives avec databricks air cancel. |
| Échecs de débogage | Extrais les journaux d'activité d’une tentative ratée et affiche l’erreur. |
| Utiliser des images Docker personnalisées | Créez et déployez une image personnalisée, puis référencez-la dans la charge de travail. |