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 vote327 views1 answer
How to create copy of all objects in list python
Python
4 votes319 views1 answer
How to calculate rmse in linear regression python
Python
0 vote376 views1 answer
Opencv loop video
Python
0 vote278 views6 answers
What is dii
Python
93 votes334 views4 answers
How to install python pyautogui
Python
33 votes284 views4 answers
Sort and reverse list in python
Python
0 vote322 views1 answer
Python how to make boxplots with swarmplot
Python
0 vote318 views1 answer
Python time-stamp conversion
Python
0 vote365 views1 answer
Python not showing in control panel but showing not installed
Python
1 vote363 views1 answer
Use incognito in selenium webdriver
Python