Groupe de commandesair
Aperçu
Cette fonctionnalité est en Aperçu public.
Le groupe de commandes air au sein de la CLI Databricks soumet et gère les charges de travail de GPU, telles que l'entraînement et l'inférence, avec AI Runtime.
These commands are not supported on Google Cloud.
databricks air run
Soumettre une charge de travail à partir d’une configuration YAML. La sortie inclut l’ID d’exécution du job et l’URL d’exécution du job.
databricks air run --file YAML_PATH [flags]
Arguments
Aucun
Options
-f, --file string
Chemin d’accès vers le YAML de configuration du workload. Cette option est requise.
--watch
Stream node 0 logs until the run reaches a terminal state. Default : false.
--override stringArray
Remplacer un champ YAML à l’aide de FIELD_PATH=VALUE. Répétez cette option pour remplacer plusieurs champs.
--dry-run
Valider la configuration sans soumettre de workload. Default : false.
--idempotency-key string
Renvoyer une exécution existante lorsque la clé a déjà été utilisée. Remplace idempotency_token dans le fichier YAML. Maximum : 64 caractères.
Exemples
L’exemple suivant soumet une charge de travail à partir de train.yaml:
databricks air run --file train.yaml
Passez un chemin de configuration comme argument distinct à -h pour afficher l'aide sur la configuration des workloads :
databricks air run -h config
databricks air run -h config.compute
databricks air run -h config.compute.accelerator_type
obtenir databricks air
Obtenir les détails d’une exécution.
databricks air get JOB_RUN_ID [flags]
Arguments
JOB_RUN_ID
ID d'exécution du job renvoyé par databricks air run.
Options
databricks air list
Lister les exécutions. Par default, la commande liste les exécutions actives soumises par l’utilisateur actuel.
databricks air list [flags]
Arguments
Aucun
Options
--limit int
Nombre maximal d’exécutions à afficher. default : 20.
--all-status
Inclure les exécutions dans des états terminaux. default : false.
--all-users
Inclure les exécutions soumises par d’autres utilisateurs. Default : false.
--filter stringArray
Filtrer les exécutions à l'aide de KEY=VALUE. Répétez cette option pour exiger plusieurs conditions. Les clés prises en charge sont accelerator_type, experiment, num_accelerators et user.
databricks air logs
Stream ou récupérer les Logs pour une exécution.
databricks air logs JOB_RUN_ID [flags]
Arguments
JOB_RUN_ID
ID d'exécution du job renvoyé par databricks air run.
Options
--node int
Sélectionner un index de nœud commençant à zéro. default : 0.
--tail int
Nombre de lignes de log existantes à afficher. default : 10000.
--minutes int
Récupérer les Logs du nombre de minutes précédent. Ne peut pas être combiné avec --tail. default : 0.
--retry int
Sélectionnez une tentative. 0 est la tentative initiale et -1 est la plus récente. Default : -1.
--download-to string
Download l’intégralité des Logs vers un répertoire. download tous les nœuds sauf si --node est défini. Ne peut pas être utilisé en même temps que --tail ou --minutes.
annulation databricks air
Annulez une ou plusieurs exécutions par ID d’exécution de job, ou annulez toutes les exécutions actives soumises par l’utilisateur actuel.
databricks air cancel [JOB_RUN_ID...] [flags]
Arguments
JOB_RUN_ID
Un ou plusieurs ID d’exécution de job. Ne peut pas être utilisé en même temps que --all.
Options
--all
Annuler toutes les exécutions actives soumises par l'utilisateur actuel. default : false.
-y, --yes
Ignorer l’invite de confirmation utilisée par --all. default : false.
databricks air pools list
Lister la capacité GPU provisionnée disponible pour le Workspace actuel.
databricks air pools list [flags]
Arguments
Aucun
Options
databricks air pools get
Obtenir les détails de la capacité de GPU provisionnement.
databricks air pools get [POOL_ID] [flags]
Arguments
POOL_ID
ID de la capacité provisionnée. L’argument est facultatif uniquement lorsque le workspace dispose d’exactement une capacité provisionnée.
Options
databricks air images push
Envoyer une image de conteneur vers Artifact Registry.
databricks air images push [flags]
Arguments
Aucun
Options
-s, --source string
L’image de conteneur source sous forme de nom avec un tag optionnel (NAME[:TAG]) ou un condensé (NAME@DIGEST). Si cette valeur est omise dans un terminal interactif, la commande invite à la saisir.
--catalog string
Le catalogue Unity Catalog de destination. Si elle est omise dans un terminal interactif, la commande demande la valeur.
--schema string
Le schéma Unity Catalog de destination. En cas d’omission dans un terminal interactif, la commande demande la valeur.
--artifact string
L’artefact de destination comme ARTIFACT[:TAG]. default, utilise le nom et le tag de l’image source, ou le tag latest lorsque la source n’a pas de tag.
--pull
Extrayez l’image source de son registre avant de la pousser au lieu de réutiliser une image locale disponible. default : false.
databricks air convert-to-dabs
Convertir une configuration YAML de charge de travail en un ensemble de ressources Databricks. Cette commande locale écrit databricks.yml et lance des fichiers sous generated_artifacts/.
Cette commande constitue une passerelle optionnelle vers un flux de travail de production avancé. Vous pouvez continuer à utiliser databricks air run sans convertir la configuration.
databricks air convert-to-dabs YAML_PATH [flags]
Arguments
YAML_PATH
Le chemin d’accès de la configuration YAML de la charge de travail.
Options
--output-dir string
Répertoire dans lequel la commande écrit le bundle. Le code source doit se trouver dans ce répertoire. default : le répertoire qui contient le fichier YAML d'entrée.
--force
Remplacer les fichiers de bundle générés qui existent déjà. default : false.
Indicateurs globaux
--debug
Whether to enable debug logging.
-h OU --help
Afficher l'aide pour la CLI Databricks, le groupe de commandes associé ou la commande associée.
--log-file string
Chaîne représentant le fichier vers lequel écrire les logs de sortie. Si cet indicateur n’est pas spécifié, l’option default consiste à écrire les logs de sortie dans stderr.
--log-format Format
Le type de format de log, text ou json. La valeur par default est text.
--log-level string
Chaîne représentant le niveau du format de log. Si non spécifié, le niveau du format de log est désactivé.
-o, --output Type
Le type de sortie de la commande, text ou json. La valeur par default est text.
-p, --profile string
Nom du profil dans le fichier ~/.databrickscfg à utiliser pour exécuter la commande. Si ce flag n’est pas spécifié et s’il existe, le profil nommé DEFAULT est utilisé.
--progress-format Format
The format to display progress logs: default, append, inplace, or json
-t, --target string
Le cas échéant, la cible du bundle à utiliser
Migrer à partir de l'ancienne CLI Python
The Databricks CLI and the deprecated Python-based air CLI use different command names and flags. Update scripts using the following mappings:
Python CLI | Databricks CLI |
|---|---|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|