pytorch use multiple gpu

Friendly Hawk answered on August 25, 2021 Popularity 10/10 Helpfulness 5/10

answer pytorch use multiple gpu

related pytorch multiple gpu

pytorch use multiple gpu

Comment

#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)

Popularity 10/10 Helpfulness 5/10 Language python

Source: pytorch.org

Tags: multiple-gpu python pytorch

Link to this answer
Share Copy Link

Contributed on Aug 25 2021

Friendly Hawk

0 Answers Avg Quality 2/10

Closely Related Answers

pytorch multiple gpu

Comment

# all GPUs
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = CreateModel()
model= nn.DataParallel(model)
model.to(device)
# specific GPUS
device = torch.device("cuda:1,3" if torch.cuda.is_available() else "cpu") ## specify the GPU id's, GPU id's start from 0.
model = CreateModel()
model= nn.DataParallel(model,device_ids = [1, 3])
model.to(device)
# OS way - programming like all GPUs
export CUDA_VISIBLE_DEVICES=1,3

Popularity 9/10 Helpfulness 8/10 Language python

Source: stackoverflow.com

Tags: multiple-gpu multiple-g

Link to this answer
Share Copy Link

Contributed on Nov 22 2022

Combative Crocodile

0 Answers Avg Quality 2/10

pytorch use multiple gpu

Contents

More Related Answers

pytorch use multiple gpu

Closely Related Answers

pytorch multiple gpu

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.