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 vote216 views2 answers
Fast output python
Python
0 vote205 views1 answer
Merge subplot matplotlib
Python
0 vote212 views2 answers
Decimal to ascii python
Python
0 vote215 views1 answer
How to fix snap "pycharm-community" has "install-snap" change in progress
Python
0 vote212 views1 answer
Pip install arcpy python 3
Python
75 votes208 views6 answers
For loop python
Python
3 votes218 views1 answer
Python split on first occurrence
Python
2 votes199 views2 answers
Generate random integer matrix python
Python
23 votes240 views6 answers
Flask post
Python Flask
3 votes208 views1 answer
For each digit in number python
Python