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