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

2 votes250 views1 answer
Best free rat for windows
Python
1 vote269 views1 answer
If none in column remove row
Python
0 vote237 views1 answer
Python running standard deviation
Python
1 vote268 views1 answer
Sin function in python does it values in radians
Python
1 vote259 views1 answer
Get number of key in dictionary python
Python
1 vote270 views1 answer
Comment choisir tout les caractère d'un str sauf les deux dernier python
Python
14 votes259 views6 answers
What is flask
Python
2 votes266 views1 answer
How to make convert numpy array to string in python
Python
10 votes258 views2 answers
Replace values in a column by condition python
Python
4 votes270 views3 answers
Flash messages django
Python Django