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

76 votes275 views6 answers
Python main
Python
0 vote294 views1 answer
Separate path python
Python
39 votes257 views6 answers
Python list contains string
Python
0 vote255 views1 answer
Limpiar consola en python
Python
0 vote272 views1 answer
How to reverse array in ruby
Python
0 vote261 views1 answer
Heatmap choos format for annotation
Python
3 votes276 views3 answers
Python - find specific name in a df
Python
0 vote295 views1 answer
Supercharged python
Python
5 votes310 views3 answers
Sys module in python
Python
0 vote305 views1 answer
Python unique values from pandas.series of sets
Python