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 vote124 views1 answer
Python Bz2 install
Python
1 vote120 views1 answer
Python to mac executable
Python
14 votes144 views3 answers
Python remove first and last character from string
Python
5 votes127 views1 answer
Getpass
Python
0 vote135 views1 answer
Pandas from range of columns
Python
0 vote123 views1 answer
Computecost pyspark
Python
0 vote153 views2 answers
How to get current latitude and longitude in python
Python
0 vote122 views1 answer
"backup-codes-username.txt
Python
1 vote142 views1 answer
Returns the first n rows
Python
0 vote136 views1 answer
Gridsearch cv
Python