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 vote341 views1 answer
Unable to locate package python-certbot-nginx
Python
5 votes354 views3 answers
Sys python
Python
-1 vote355 views1 answer
Set label text size tkinter
Python
4 votes420 views1 answer
Django is null
Python Django
1 vote321 views1 answer
Python find smallest value in 2d list
Python
3 votes363 views6 answers
Concatenate List
Python
2 votes321 views1 answer
String module in python
Python
62 votes337 views6 answers
Os get current directory
Python
12 votes362 views5 answers
Python turtle documentation
Python
0 vote326 views1 answer
Python tuple range
Python