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 vote109 views1 answer
Df reanme columns
Python
1 vote94 views1 answer
Create period pandas
Python
1 vote145 views1 answer
Unresolved reference 'django'
Python
0 vote97 views1 answer
How to load user from jwt token request django
Python
0 vote105 views1 answer
Python pandas return column name of a specific column
Python
2 votes162 views5 answers
Swap keys and values in dictionary python
Python
1 vote94 views1 answer
Python - count the frquency of a vlaue in a coulmn
Python
0 vote109 views1 answer
Unit converter idle python
Python
0 vote117 views1 answer
Geopandas españa map
Python
4 votes94 views3 answers
Python os
Python