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

10 votes111 views5 answers
Docker mount volume
Python
0 vote109 views1 answer
Python get file path from in os.walk
Python
-1 vote125 views1 answer
Add column python list
Python
0 vote106 views1 answer
Two input string sum in django
Python
-1 vote113 views2 answers
Sum of all nan values pandas
Python
0 vote134 views1 answer
Flask print request headers
Python Flask
2 votes106 views1 answer
String module in python
Python
0 vote105 views2 answers
Flask upload file to s3
Python
6 votes113 views2 answers
If ,elif, else in python
Python
12 votes122 views4 answers
Remove nan particular column pandas
Python