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 vote109 views1 answer
Remote python running line by line visual code
Python
4 votes106 views1 answer
How to import reverse
Python
0 vote98 views1 answer
Subplots whitespace
Python
1 vote115 views3 answers
Discord py aliases
Python
0 vote116 views2 answers
Pyton como identificar se é numero
Python
1 vote138 views1 answer
How to get images on flask page
Python Flask
0 vote103 views1 answer
Statsmodels fitted values
Python
4 votes102 views6 answers
If statement python
Python
1 vote110 views1 answer
Datetime not defined python
Python
12 votes107 views6 answers
Print string and variable python
Python