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 vote194 views1 answer
Length of number without using len function
Python
93 votes189 views6 answers
Creating virtual environment python
Python
1 vote200 views2 answers
What is a cube plus b cube
Python
0 vote213 views1 answer
From matrix to array python
Python
0 vote250 views1 answer
Enumerate in django templte
Python Django
0 vote244 views2 answers
Django rest framework authentication
Python Django
1 vote184 views2 answers
Python for looop array value and index
Python
0 vote179 views1 answer
Pysft connection drop issue
Python
0 vote183 views1 answer
Azure functions read only file system
Python
0 vote186 views1 answer
Python counting subfolders on specific level
Python