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 votes130 views2 answers
Cv2 add text
Python
1 vote148 views1 answer
List count frequency python
Python
8 votes138 views6 answers
How to convert categorical data to numerical data in python
Python
8 votes94 views2 answers
Python "is"
Python
1 vote136 views1 answer
Pylint: disable=unused-argument
Python
0 vote139 views2 answers
Message: 'geckodriver' executable needs to be in PATH.
Python
0 vote149 views1 answer
Search number is complete or no python
Python
0 vote140 views1 answer
How to subtract numbers in python
Python
0 vote132 views1 answer
Kivy lang
Python
1 vote150 views1 answer
Get time between things python
Python