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 vote168 views1 answer
Append to list in dictionary python if exists
Python
1 vote159 views1 answer
How to create routes in flask
Python
1 vote218 views1 answer
Factorial sequence code in python with while loops
Python
2 votes171 views1 answer
Python check is os is windows
Python
2 votes164 views2 answers
How to get key of a particular value in dictionary python using index
Python
0 vote164 views1 answer
Passport ocr python
Python
0 vote168 views2 answers
How to install python pip in ubuntu
Python
8 votes169 views1 answer
Use beautifulsoup
Python
0 vote160 views1 answer
Which clause is used to place condition with GROUP BY clause in a table
Python
6 votes160 views1 answer
Get all combinations from two lists python
Python