사용
설명
docker run을 래핑하고 W&B 환경 변수를 자동으로 주입합니다.
다음을 주입하면서 모든 인수를 docker run에 그대로 전달합니다:
-
WANDB_API_KEY: 로그인되어 있으면 현재 API 키를 주입하여
컨테이너가 W&B에 인증할 수 있도록 합니다.
- WANDB_DOCKER: 인수에서 이미지를 감지할 수 있으면 확인된 이미지 ID를 주입하여 어떤 도커 이미지가 해당 run을 생성했는지 W&B가 추적할 수 있도록 합니다.
nvidia-docker가 감지되고 --runtime이 아직 설정되지 않은 경우
--runtime nvidia를 자동으로 추가합니다.
예를 들어, “my-image” 컨테이너 내부에서 Python 스크립트 train.py를 실행하려면:
인수
| 이름 | 기본값 | 유형 |
|---|---|---|
docker_run_args | STR | False |