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 vote289 views1 answer
Add a constant to a list python
Python
6 votes285 views2 answers
Convert negative to positive in python
Python
2 votes307 views1 answer
Argparse example python pyimagesearch
Python
10 votes368 views1 answer
Add static file in django
Python Django
0 vote296 views1 answer
Read data from excel file using openbyxl
Python
0 vote290 views1 answer
Python sum of 10 numbers from user input
Python
0 vote311 views2 answers
Python declare 2d array size
Python
0 vote282 views1 answer
Remove words from set if in list python site:stackoverflow.com
Python
88 votes283 views6 answers
How to make a loop in python
Python
0 vote316 views1 answer
Flask get uploaded file size
Python