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 vote395 views1 answer
Coefficient of variation python
Python
1 vote352 views1 answer
Import open3d Illegal instruction (core dumped)
Python
0 vote331 views1 answer
Returns the dataframe with the modified Title column in which the updated groupings are reflected.
Python
0 vote355 views1 answer
Django create object with default today date
Python
1 vote343 views1 answer
Cast tensor type pytorch
Python
11 votes374 views4 answers
Invert a dictionary python
Python
0 vote341 views1 answer
Runtime errors in python
Python
7 votes359 views3 answers
Pytorch load model
Python
5 votes364 views3 answers
Show multiple plots python
Python
3 votes385 views2 answers
How to find word in file python
Python