Dockerfile を指定して、 W&B に Docker イメージをビルドさせることはできますか？

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

マウントを使用するように Dockerfile をフォーマットしてください。詳細については、Docker 公式ドキュメントの Mounts に関するドキュメント を参照してください。

Dockerfile の設定が完了したら、以下の 3 つのいずれかの方法で W&B に指定します。

• 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"
}