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

27 votes235 views2 answers
Draw a line pygame
Python
0 vote222 views1 answer
D-tale colab
Python
2 votes218 views3 answers
Fibonacci formula python
Python
9 votes229 views2 answers
Fill missing values in column pandas with mean
Python
0 vote255 views1 answer
Pandas describe kurtosis skewness
Python
1 vote223 views1 answer
Keras auc without tf.metrics.auc
Python
0 vote242 views1 answer
Importing time and sleep. python
Python
0 vote235 views1 answer
Add text to axis
Python
0 vote254 views1 answer
Python string: .find()
Python
6 votes227 views3 answers
Python get dictionary keys
Python