W&B に Dockerfile を指定して、Docker イメージを作成してもらうことはできますか？

この機能は、要件が安定しているがコードベースが頻繁に変化するプロジェクトに適しています。

Dockerfile をマウントとして使用するようにフォーマットしてください。詳細は、Docker Docs の Mounts ドキュメントをご覧ください。

Dockerfile を設定した後、W&B に指定する方法は次の3つです:

• Dockerfile.wandb を使用する
• W&B CLI を使用する
• W&B App を使用する

Dockerfile.wandb

W&B run のエントリポイントと同じディレクトリーに Dockerfile.wandb ファイルを含めます。W&B はこのファイルを組み込みの Dockerfile ではなく使用します。

W&B CLI

wandb launch コマンドに --dockerfile フラグを使用してジョブをキューに追加します:

wandb launch --dockerfile path/to/Dockerfile

W&B app

W&B App でジョブをキューに追加する際、Overrides セクションで Dockerfile のパスを指定します。それをキーと値のペアとして入力し、キーを "dockerfile"、値を Dockerfile のパスとします。

次の JSON は、ローカルディレクトリーに Dockerfile を含む方法を示しています:

Launch job W&B App

{
  "args": [],
  "run_config": {
    "lr": 0,
    "batch_size": 0,
    "epochs": 0
  },
  "entrypoint": [],
  "dockerfile": "./Dockerfile"
}