Pytorch use multiple gpu

#easiest solution is to wrap you model in DataParallel like so:

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = Model(input_size, output_size)
if torch.cuda.device_count() > 1:
  print("Let's use", torch.cuda.device_count(), "GPUs!")
  model = nn.DataParallel(model)

model.to(device)

Source: pytorch.org

Similar codes

34 votes335 views3 answers
Python add command
Python
3 votes214 views1 answer
Randonm forest python
Python
18 votes279 views6 answers
Python odd or even
Python
11 votes325 views6 answers
Pandas group by count
Python
0 vote301 views1 answer
How to make commas appear in integers in terminal python
Python
1 vote274 views1 answer
Simple way of finding file extension python programming
Python
0 vote287 views1 answer
Pandas gropu by
Python
0 vote285 views1 answer
Creating new column with dictionary
Python
9 votes287 views1 answer
Upgrade python version mc
Python
0 vote316 views1 answer
Python change terminal name
Python