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

12 votes112 views1 answer
Link shortener
Python
0 vote130 views1 answer
Inpunt and output
Python
0 vote109 views1 answer
Dot operator in python
Python
0 vote125 views1 answer
Python - retrieve unconnected node pairs
Python
0 vote117 views2 answers
User logout in django rest framework
Python
0 vote106 views1 answer
Converting string key to int py
Python
0 vote108 views1 answer
Ordenar lista python
Python
-1 vote110 views1 answer
Starry spheres
Python
1 vote105 views1 answer
Pandas merge dataframes from a list
Python
0 vote114 views1 answer
Python declare immutable variable
Python