참고
Click here to download the full example code
멀티-GPU 예제¶
데이터 병렬 처리(Data Parallelism)는 미니-배치를 여러 개의 더 작은 미니-배치로 자르고 각각의 작은 미니배치를 병렬적으로 연산하는 것입니다.
데이터 병렬 처리는 torch.nn.DataParallel 을 사용하여 구현합니다.
DataParallel 로 감쌀 수 있는 모듈은 배치 차원(batch dimension)에서
여러 GPU로 병렬 처리할 수 있습니다.
DataParallel¶
import torch
import torch.nn as nn
class DataParallelModel(nn.Module):
def __init__(self):
super().__init__()
self.block1 = nn.Linear(10, 20)
# wrap block2 in DataParallel
self.block2 = nn.Linear(20, 20)
self.block2 = nn.DataParallel(self.block2)
self.block3 = nn.Linear(20, 20)
def forward(self, x):
x = self.block1(x)
x = self.block2(x)
x = self.block3(x)
return x
CPU 모드인 코드를 바꿀 필요가 없습니다.
DataParallel에 대한 문서는 여기 에서 확인하실 수 있습니다.
래핑된 모듈의 속성
모듈을 DataParallel 로 감싼 후에는 모듈의 속성(예. 사용자 정의 메소드)에
접근할 수 없게 됩니다. 이는 DataParallel 이 몇몇 새로운 멤버를 정의하기 때문에
다른 속성에 접근을 허용하는 것이 충돌을 일으킬 수도 있기 때문입니다.
그래도 속성에 접근하고자 한다면 아래와 같이 DataParallel 의 서브클래스를
사용하는 것이 좋습니다.
class MyDataParallel(nn.DataParallel):
def __getattr__(self, name):
return getattr(self.module, name)
DataParallel이 구현된 기본형(Primitive):
일반적으로, PyTorch의 nn.parallel 기본형은 독립적으로 사용할 수 있습니다. 간단한 MPI류의 기본형을 구현해보겠습니다:
복제(replicate): 여러 기기에 모듈을 복제합니다.
분산(scatter): 첫번째 차원에서 입력을 분산합니다.
수집(gather): 첫번째 차원에서 입력을 수집하고 합칩니다.
병렬적용(parallel_apply): 이미 분산된 입력의 집합을 이미 분산된 모델의 집합에 적용합니다.
더 명확히 알아보기 위해, 위 요소 사용하여 구성한 data_parallel
함수를 살펴보겠습니다.
def data_parallel(module, input, device_ids, output_device=None):
if not device_ids:
return module(input)
if output_device is None:
output_device = device_ids[0]
replicas = nn.parallel.replicate(module, device_ids)
inputs = nn.parallel.scatter(input, device_ids)
replicas = replicas[:len(inputs)]
outputs = nn.parallel.parallel_apply(replicas, inputs)
return nn.parallel.gather(outputs, output_device)
모델의 일부는 CPU, 일부는 GPU에서¶
일부는 CPU에서, 일부는 GPU에서 신경망을 구현한 짧은 예제를 살펴보겠습니다
device = torch.device("cuda:0")
class DistributedModel(nn.Module):
def __init__(self):
super().__init__(
embedding=nn.Embedding(1000, 10),
rnn=nn.Linear(10, 10).to(device),
)
def forward(self, x):
# CPU에서 연산합니다.
x = self.embedding(x)
# GPU로 보냅니다.
x = x.to(device)
# GPU에서 연산합니다.
x = self.rnn(x)
return x
지금까지 기존 Torch 사용자를 위한 간단한 PyTorch 개요를 살펴봤습니다. 배울 것은 아주 많이 있습니다.
optim 패키지, 데이터 로더 등을 소개하고 있는 더 포괄적인 입문용 튜토리얼을
보시기 바랍니다: PyTorch로 딥러닝하기: 60분만에 끝장내기.
또한, 다음의 내용들도 살펴보세요.
Total running time of the script: ( 0 minutes 0.001 seconds)