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

この機能は、依存関係は安定している一方で、コードベースが頻繁に変わるプロジェクトに適しています。

Dockerfile はマウントを使用する形式で記述してください。詳細は、Docker Docs の 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"
}