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 vote272 views1 answer
Video steganography using python
Python
0 vote317 views2 answers
Update a package with pip in django
Python Django
-1 vote284 views1 answer
Convert .tif images files to .jpeg in python
Python
0 vote304 views1 answer
_csv.Error: field larger than field limit (131072)
Python
0 vote248 views1 answer
2D array questions python
Python
-1 vote253 views1 answer
Groupby
Python
0 vote268 views1 answer
Pillow rgb to grayscale
Python
1 vote285 views2 answers
AttributeError: module 'yaml' has no attribute 'load_all'
Python
0 vote272 views2 answers
Fix ImportError: No module named PIL
Python
0 vote273 views1 answer
Python String with Alphanumeric characters
Python