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

-2 votes323 views5 answers
Convert python to java
Python
0 vote363 views1 answer
Django action when create model
Python Django
0 vote301 views1 answer
Add percentage column pandas
Python
1 vote301 views2 answers
Python line break inside string
Python
1 vote299 views1 answer
Cprofile usage python
Python
0 vote352 views1 answer
Firebase python flask
Python Flask
0 vote328 views1 answer
Django model different schema
Python
10 votes313 views2 answers
Python re.search()
Python
0 vote315 views1 answer
Get the positions of items of ser2 in ser1 as a list python
Python
0 vote290 views1 answer
List alpha numeric
Python