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 vote287 views1 answer
Smtplib send caleneder email
Python
1 vote265 views1 answer
Django get model class from queryset
Python
1 vote279 views1 answer
Log loss python
Python
14 votes285 views2 answers
Two groupby pandas
Python
9 votes265 views1 answer
Make tkinter btn disable
Python
0 vote265 views1 answer
All python commands list and what they do
Python
27 votes269 views4 answers
Create a dataframe from dict
Python
2 votes275 views1 answer
Python sympy symbols
Python
4 votes265 views1 answer
Python windows hide files
Python
2 votes276 views1 answer
Get local ipv4
Python