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

0 vote41 views1 answer
Python regex remove digits from string
Python
0 vote47 views1 answer
Icon tkiner
Python
3 votes48 views2 answers
Count number of each item in list python
Python
0 vote79 views1 answer
Numpy init array
Python
20 votes44 views2 answers
Create a list of a certain length python
Python
0 vote49 views1 answer
Print Wavelet modes
Python
0 vote43 views1 answer
Get value of 1 field in model django
Python
0 vote40 views1 answer
Set select group of columns to numeric pandas
Python
0 vote40 views1 answer
MovieWriter stderr: ffmpeg: error while loading shared libraries: libopenh264.so.5: cannot open shared object file: No such file or directory
Python
0 vote48 views1 answer
Tusha
Python