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 vote160 views2 answers
Wav file to array python
Python
0 vote166 views1 answer
Pycharm Error: Please select a valid Python interpreter
Python
3 votes132 views3 answers
Flask migrate install
Python
3 votes144 views3 answers
Python iterar claves
Python
0 vote137 views1 answer
Python matplotlib pyplot set axis equals
Python
0 vote132 views1 answer
Read page source from text file python
Python
40 votes209 views6 answers
Plt size
Python
0 vote174 views1 answer
Python get last cell value
Python
1 vote137 views1 answer
Python datetime tomorrow date
Python
0 vote145 views1 answer
Easygui text adventure in python 3
Python