Can I specify a Dockerfile and let W&B build a Docker image for me?

이 기능은 요구 사항은 안정적이지만 코드 베이스가 자주 변경되는 프로젝트에 적합합니다.

Dockerfile을 구성한 후에는 다음 세 가지 방법 중 하나로 W&B에 지정합니다.

  • Dockerfile.wandb 사용
  • W&B CLI 사용
  • W&B App 사용

W&B run 의 진입점과 동일한 디렉토리에 Dockerfile.wandb 파일을 포함합니다. W&B는 내장된 Dockerfile 대신 이 파일을 사용합니다.

wandb launch 코맨드와 함께 --dockerfile 플래그를 사용하여 작업을 대기열에 추가합니다.

wandb launch --dockerfile path/to/Dockerfile

W&B App에서 대기열에 작업을 추가할 때 Overrides 섹션에서 Dockerfile 경로를 제공합니다. "dockerfile" 을 키 로, Dockerfile의 경로를 값 으로 하여 키-값 쌍으로 입력합니다.

다음 JSON은 로컬 디렉토리에 Dockerfile을 포함하는 방법을 보여줍니다.

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