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 vote153 views1 answer
Argsort in descending order numpy
Python
0 vote157 views1 answer
Random ordered slice of an array
Python
0 vote164 views1 answer
Python pow
Python
-1 vote147 views1 answer
How to separate numeric and categorical variables in python
Python
0 vote154 views1 answer
Python second interval
Python
9 votes185 views2 answers
Python check if number is in range
Python
3 votes214 views6 answers
ImportError: Couldn't import Django. Are you sure it's installed and available on your PYTHONPATH environment variable? Did you forget to activate a virtual environment?
Python Django
0 vote139 views2 answers
Change markersize in legend matplotlib
Python
0 vote178 views1 answer
Django from
Python Django
-1 vote224 views3 answers
ValueError: If using all scalar values, you must pass an index
Python