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 vote142 views1 answer
Python find oldest and newest date
Python
2 votes225 views2 answers
For each loop python 3
Python
0 vote151 views1 answer
Comparison python
Python
81 votes177 views6 answers
Check python version
Python
0 vote141 views2 answers
How to check if a string ends with a substring python
Python
1 vote147 views2 answers
Python copy variable
Python
6 votes215 views4 answers
Python Day of the week
Python
0 vote180 views1 answer
How to make python open a program/desktop app
Python
4 votes154 views2 answers
Discord py color
Python
0 vote139 views1 answer
Print all alphabets from a to z in python
Python