wandb docker

사용

docker [OPTIONS] [DOCKER_RUN_ARGS]... [DOCKER_IMAGE]

설명

W&B가 설정된 도커 컨테이너에서 코드를 실행합니다. 도커 컨테이너를 시작하고 WANDB_DOCKERs 환경 변수를 주입하며, 기본적으로 현재 작업 디렉터리를 /app에 마운트합니다. 로그인한 경우 WANDB_API_KEY도 주입합니다. wandb가 설치되도록 컨테이너 엔트리포인트를 재정의합니다. 추가 인수를 전달하면 이미지 이름 앞의 docker run에 삽입됩니다. 이미지를 지정하지 않으면 기본 이미지를 사용합니다. --jupyter를 사용하면 8888 포트에서 JupyterLab을 설치하고 시작합니다. 호스트에서 NVIDIA 도커를 사용할 수 있으면 NVIDIA 런타임을 자동으로 활성화합니다. 호스트 머신에 도커가 설치되어 있고 실행 중이어야 합니다. 기존 docker run 명령어의 엔트리포인트를 수정하지 않고 W&B 환경 변수를 주입하려면 wandb docker-run을 사용하세요. 예를 들어, 기본 이미지를 실행하고 데이터셋을 컨테이너에 마운트하려면 다음과 같이 하세요:

wandb docker -v /mnt/dataset:/app/data

기본 이미지를 실행하고 JupyterLab을 시작하려면:

wandb docker -v /mnt/dataset:/app/data --jupyter

트레이닝 명령을 사용해 GPU 지원 이미지를 실행하려면:

wandb docker wandb/deepo:keras-gpu --no-tty --cmd "python train.py"

인수

이름	기본값	유형
`docker_run_args`	STR	False
`docker_image`	STR	False

옵션

플래그	유형	설명
`--nvidia`	불리언 플래그	NVIDIA 런타임을 사용합니다. `nvidia-docker`가 있으면 기본값은 `nvidia`입니다. 기본값: False
`--digest`	불리언 플래그	이미지 다이제스트를 출력한 후 종료합니다. 기본값: False
`--jupyter`	불리언 플래그	컨테이너에서 JupyterLab을 실행합니다. 기본값: False
`--dir`	STR	컨테이너에서 코드를 마운트할 디렉터리입니다. 기본값: /app
`--no-dir`	불리언 플래그	현재 디렉터리를 마운트하지 않습니다. 기본값: False
`--shell`	STR	컨테이너 시작 시 사용할 셸입니다. 기본값: /bin/bash
`--port`	STR	Jupyter를 바인딩할 호스트 포트입니다. 기본값: 8888
`--cmd`	STR	컨테이너에서 실행할 명령어입니다. 기본값: None
`--no-tty`	불리언 플래그	TTY 없이 명령어를 실행합니다. 기본값: False

W&B Models

W&B Weave

Serverless Inference

Serverless Training

사용

설명

인수

옵션

​사용

​설명

​인수

​옵션

사용

설명

인수

옵션