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 vote239 views1 answer
Python REPL loop
Python
1 vote204 views1 answer
Split string and convert to int python
Python
0 vote177 views1 answer
2checkout python
Python
0 vote205 views3 answers
Autoextract scrapy spider
Python
0 vote203 views1 answer
Left fill with zeros
Python
1 vote196 views3 answers
Count function in python
Python
1 vote183 views1 answer
Tqdm parallel
Python
49 votes199 views6 answers
How to create dataframe in python
Python
0 vote188 views1 answer
Python can't add int on string
Python
9 votes198 views3 answers
How to detect when a key is pressed in pygame
Python