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 votes187 views3 answers
Mysql connection python
Python
0 vote243 views1 answer
Python logging basicConfig+time
Python
-1 vote201 views1 answer
Multiple assignment python
Python
4 votes194 views2 answers
How to change turtle shape in python
Python
17 votes227 views5 answers
What is python used for
Python
5 votes208 views5 answers
Formatted string in python
Python
0 vote196 views1 answer
Boto3 read excel file from s3 into pandas
Python
1 vote192 views1 answer
Django ckeditor not working
Python
0 vote195 views1 answer
Circular heatmap python
Python
0 vote252 views1 answer
How to store a int value in django sessions
Python Django