wandb docker

코드를 Docker 컨테이너에서 실행하세요.

W&B Docker를 사용하면 wandb가 올바르게 구성된 상태로 Docker 이미지에서 코드를 실행할 수 있습니다. 이 도구는 컨테이너에 WANDB_DOCKER 및 WANDB_API_KEY 환경 변수를 추가하고, 기본적으로 현재 디렉터리를 /app에 마운트합니다. 추가 인수를 전달할 수 있으며, 이 인수들은 이미지 이름이 선언되기 전에 docker run 명령에 추가됩니다. 이미지를 지정하지 않으면 기본 이미지를 자동으로 선택합니다:

wandb docker -v /mnt/dataset:/app/data
wandb docker gcr.io/kubeflow-images-public/tensorflow-1.12.0-notebook-cpu:v0.4.0 --jupyter
wandb docker wandb/deepo:keras-gpu --no-tty --cmd "python train.py --epochs=5"

기본적으로 entrypoint를 재정의하여 wandb가 설치되어 있는지 확인하고, 없으면 설치합니다. —jupyter 플래그를 전달하면 jupyter가 설치되어 있는지 확인하고 포트 8888에서 jupyter lab을 시작합니다. 시스템에서 nvidia-docker를 감지하면 nvidia 런타임을 사용합니다. wandb가 기존 docker run 명령에 사용할 환경 변수만 설정하도록 하고 싶다면 wandb docker-run 명령을 참고하세요.

​

사용 방법

wandb docker [DOCKER_RUN_ARGS] [DOCKER_IMAGE] [OPTIONS]

​

인자

인자	설명	필수 여부
DOCKER_RUN_ARGS	설명이 제공되지 않습니다	아니요
DOCKER_IMAGE	설명이 제공되지 않습니다	아니요

​

옵션

옵션	설명
--nvidia	nvidia 런타임을 사용합니다. nvidia-docker가 있으면 기본 런타임은 nvidia입니다 (기본값: False)
--digest	이미지 다이제스트를 출력하고 종료합니다 (기본값: False)
--jupyter	컨테이너에서 Jupyter Lab을 실행합니다 (기본값: False)
--dir	컨테이너에서 코드를 마운트할 디렉터리를 지정합니다 (기본값: /app)
--no-dir	현재 디렉터리를 마운트하지 않습니다 (기본값: False)
--shell	컨테이너를 시작할 때 사용할 셸을 지정합니다 (기본값: /bin/bash)
--port	Jupyter가 바인딩될 호스트 포트를 지정합니다 (기본값: 8888)
--cmd	컨테이너에서 실행할 명령을 지정합니다
--no-tty	TTY 없이 명령을 실행합니다 (기본값: False)