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