메인 콘텐츠로 건너뛰기
멀티코어 또는 멀티 GPU 머신에서 W&B Sweep 에이전트를 병렬화합니다. 시작하기 전에 W&B Sweep이 초기화되어 있는지 확인하세요. W&B Sweep을 초기화하는 방법에 대한 자세한 내용은 Sweeps 초기화를 참조하세요.

멀티 CPU 머신에서 병렬화

사용 사례에 따라 아래 탭에서 CLI 또는 Jupyter Notebook 내에서 W&B Sweep 에이전트를 병렬화하는 방법을 확인하세요.
터미널에서 wandb agent command를 사용해 여러 CPU에서 sweep 에이전트를 병렬화하세요. sweep를 초기화할 때 반환된 sweep ID를 입력하세요.
  1. 로컬 머신에서 터미널 창을 두 개 이상 엽니다.
  2. 아래 코드 스니펫을 복사해 붙여넣고 sweep_id를 본인의 sweep ID로 바꾸세요:
wandb agent sweep_id

다중 GPU 머신에서 병렬 처리하기

CUDA Toolkit을 사용하는 터미널에서 W&B Sweep 에이전트를 여러 GPU에 걸쳐 병렬로 실행하려면 다음 절차를 따르세요.
  1. 로컬 머신에서 터미널 창을 두 개 이상 엽니다.
  2. W&B Sweep 작업(wandb agent)을 시작할 때 CUDA_VISIBLE_DEVICES를 사용해 사용할 GPU 인스턴스를 지정합니다. 사용할 GPU 인스턴스에 해당하는 정수 값을 CUDA_VISIBLE_DEVICES에 할당합니다.
예를 들어, 로컬 머신에 NVIDIA GPU가 두 개 있다고 가정해 보겠습니다. 터미널 창을 열고 CUDA_VISIBLE_DEVICES0으로 설정합니다(CUDA_VISIBLE_DEVICES=0). 다음 예시에서 sweep_ID를 W&B Sweep을 초기화할 때 반환되는 W&B Sweep ID로 바꾸세요. 터미널 1
CUDA_VISIBLE_DEVICES=0 wandb agent sweep_ID
두 번째 터미널 창을 엽니다. CUDA_VISIBLE_DEVICES1로 설정합니다(CUDA_VISIBLE_DEVICES=1). 아래 코드 스니펫에 나오는 sweep_ID에 동일한 W&B Sweep ID를 붙여 넣습니다: 터미널 2
CUDA_VISIBLE_DEVICES=1 wandb agent sweep_ID