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

1 vote137 views1 answer
Can list comprehenios contain else
Python
1 vote141 views1 answer
Transparancy argument pyplot
Python
0 vote151 views1 answer
How to username in python?
Python
0 vote130 views1 answer
Python get nearest value in list
Python
5 votes147 views1 answer
Godot enum
Python
-2 votes140 views1 answer
Tkinter bind function with arguments
Python
95 votes136 views6 answers
Create virtual env
Python
0 vote148 views1 answer
Python as-lookup
Python
1 vote150 views1 answer
Pygame get if button is released
Python
0 vote140 views1 answer
Appending objects to a list contained in a dictionary python
Python