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

0 vote196 views1 answer
Python get list of file and time created
Python
0 vote287 views1 answer
2--2 in python prints?
Python
0 vote185 views3 answers
NameError: name ‘np’ is not defined
Python
1 vote195 views1 answer
Flask api with parameter
Python
0 vote184 views1 answer
Number of unique pairs in columns pandas
Python
0 vote191 views1 answer
Check if argv exists python
Python
0 vote214 views1 answer
Amazing Trees with python turtle
Python
14 votes305 views2 answers
Amazon cli on commadline
Python
0 vote214 views1 answer
How to add an symbol to a certain part of a list python
Python
0 vote225 views1 answer
Django default template location
Python