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

4 votes343 views2 answers
Median of a list python
Python
2 votes352 views2 answers
Speech to text
Python
0 vote311 views1 answer
Read a file and split the words python
Python
12 votes347 views2 answers
Read excel file spyder
Python
0 vote360 views1 answer
Django trigger when an instance od data is deleted from model
Python Django
9 votes448 views2 answers
Flask how to run app
Python Flask
0 vote350 views1 answer
Render django views
Python
0 vote346 views1 answer
Python flatten one liner
Python
1 vote354 views1 answer
Cv2 draw box
Python
2 votes390 views1 answer
Urlpatterns += static(r'^static/(?P<path>.*)$', 'django.views.static.serve', {'document_root': settings.STATIC_ROOT})
Python Django