メインコンテンツまでスキップ

AI Runtime CLI

備考

プレビュー

この機能は パブリック プレビュー段階です。

air コマンドラインインターフェースは、分散トレーニングワークロードを AI Runtime (オンデマンドのサーバレス GPU コンピュートプラットフォーム) 上で提出および管理を行います。CLI は YAML ベースのジョブ構成を使用し、MLflow と統合し、ワークスペースベースおよび Git ベースのコードワークフローをサポートします。

air CLIはAIランタイムと同じコンプライアンス標準に準拠しています。サポートされている標準と例外については、「コンプライアンスセキュリティプロファイル」を参照してください。

CLIの使用方法

AI Runtime CLI を使用したい場合:

  • ノートブックを開かずに、ラップトップやコードエディタからGPUトレーニングワークロードを提出できます。
  • トレーニングジョブをYAMLで宣言的に定義することで、ソース管理に登録できます。
  • インタラクティブセッションよりも長く実行する必要がある、または複数のノードにまたがるワークロードである長時間実行のトレーニングジョブやマルチノード分散トレーニングを実行します。

ノートブック内Python API(@distributedおよび@ray_launch)については、代わりにノートブックでの分散トレーニングを参照してください。

ワークロードを送信するのではなく、ターミナルまたは IDE から SSH 経由で単一 GPU ノードで対話的に作業するには、databricks ssh connect を使用します。SSHトンネルを使用してDatabricksに接続するを参照してください。

このセクションの内容: