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