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

AI Runtime CLI

備考

ベータ版

AI Runtime CLI は ベータ版です。

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

CLIの使用方法

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

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

ノートブック内のPython API(@distributedおよび@ray_launch)については、代わりにマルチGPUワークロードを参照してください。

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

このセクションの内容: