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 vote155 views1 answer
Python serial port
Python
0 vote135 views2 answers
Convert python code to java using jython
Python
5 votes132 views1 answer
Python leap year solution
Python
0 vote131 views1 answer
How to add the sum of multiple columns into another column in a dataframe
Python
0 vote182 views1 answer
Odoo site map for employees hierarchy
Python
1 vote133 views1 answer
Constructor python variables
Python
5 votes135 views1 answer
Tkinter entry default value
Python
2 votes153 views3 answers
Scatter plot actual vs predicted python
Python
0 vote97 views1 answer
Python3 delete file
Python
0 vote133 views1 answer
Add header info in django response
Python