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 vote336 views1 answer
Add sign to y axis values python
Python
1 vote330 views1 answer
Python print os platform
Python
1 vote305 views1 answer
Python - count the frquency of a vlaue in a coulmn
Python
0 vote534 views1 answer
Python devops modules
Python
0 vote372 views1 answer
Suppress pandas future warnings
Python
2 votes330 views3 answers
Combine list of dicts
Python
0 vote311 views2 answers
Logging the terminal output to a file
Python
9 votes321 views2 answers
Merge 2 dataframes pythom
Python
0 vote332 views1 answer
Pycharm flask invalid or missing encoding declaration
Python
0 vote322 views1 answer
Async, await, python
Python