search

razvancaramalau / Sequential-GCN-for-Active-Learning

50 stars 17 forks source link

Using data parallel and custom gpus #6

Open manza-ari opened 2 years ago

manza-ari commented 2 years ago

How can I use multiple GPUs for this code? like data-parallel?

razvancaramalau commented 2 years ago

you can definitely add the wrappers of nn.DataParallel() to the learner definition in main (and for the GCN in the selection_methods). Make sure to add your available devices and push the data on the first GPU.