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

7 votes144 views2 answers
Python string prefix
Python
1 vote233 views1 answer
Using bs4 to obtain html element by id
Python
1 vote172 views1 answer
Does jupyter notebook need internet
Python
26 votes161 views1 answer
Python gmail
Python
0 vote171 views1 answer
Read file from s3 python
Python
9 votes171 views1 answer
Python delete none from list
Python
9 votes169 views1 answer
Save an image in python as grayscale cv2
Python
2 votes172 views4 answers
Difference between 2 dataframes
Python
3 votes186 views2 answers
Find minimum and maximum element in an array
Python
0 vote233 views1 answer
Flask put request example
Python Flask