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

2 votes224 views2 answers
Python get file size in mb
Python
4 votes219 views2 answers
Pandas groupby mean
Python
33 votes213 views4 answers
Change dataframe column type
Python
2 votes231 views3 answers
Python find first occurrence in list
Python
0 vote206 views1 answer
Say hello to someone in python
Python
1 vote224 views1 answer
Return result from exec python
Python
33 votes230 views6 answers
Set in python
Python
0 vote248 views1 answer
Selenium press tab python
Python
0 vote211 views1 answer
Ocaml returns the last element of a list
Python
1 vote202 views1 answer
Required validator python WTForms
Python