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 votes162 views1 answer
Hibernate windows with python
Python
0 vote166 views1 answer
Boto3 read excel file from s3 into pandas
Python
39 votes172 views6 answers
Pandas drop rows with nan in a particular column
Python
0 vote180 views1 answer
Custom pylatex command
Python
2 votes159 views2 answers
Area of trapezium
Python
1 vote192 views1 answer
Number field in django
Python
0 vote248 views1 answer
Django forms request
Python Django
0 vote173 views2 answers
How to round to 3 significant figures in python
Python
0 vote190 views2 answers
Get random vowel python
Python
1 vote204 views1 answer
How to restart loop python
Python