멀티 CPU 머신에서 병렬화
- CLI
- Jupyter Notebook
터미널에서
wandb agent command를 사용해 여러 CPU에서 sweep 에이전트를 병렬화하세요. sweep를 초기화할 때 반환된 sweep ID를 입력하세요.- 로컬 머신에서 터미널 창을 두 개 이상 엽니다.
- 아래 코드 스니펫을 복사해 붙여넣고
sweep_id를 본인의 sweep ID로 바꾸세요:
다중 GPU 머신에서 병렬 처리하기
- 로컬 머신에서 터미널 창을 두 개 이상 엽니다.
- W&B Sweep 작업(
wandb agent)을 시작할 때CUDA_VISIBLE_DEVICES를 사용해 사용할 GPU 인스턴스를 지정합니다. 사용할 GPU 인스턴스에 해당하는 정수 값을CUDA_VISIBLE_DEVICES에 할당합니다.
CUDA_VISIBLE_DEVICES를 0으로 설정합니다(CUDA_VISIBLE_DEVICES=0). 다음 예시에서 sweep_ID를 W&B Sweep을 초기화할 때 반환되는 W&B Sweep ID로 바꾸세요.
터미널 1
CUDA_VISIBLE_DEVICES를 1로 설정합니다(CUDA_VISIBLE_DEVICES=1). 아래 코드 스니펫에 나오는 sweep_ID에 동일한 W&B Sweep ID를 붙여 넣습니다:
터미널 2