Multiple GPUs

[PratyushAvi]

Hi,

I'm trying to set up MXMNet to use multiple GPUs.
I've changed

model = MXMNet(config).to(device)

to

model = MXMNet(config)
model = nn.DataParallel(model)
model.to(device)

However, this appears to be insufficient. Is there anything else or something else I need to do to use multiple GPUs?

Thank you!

[zetayue]

Since we are using PyTorch Geometric to build our model, you may need to use the related options provided by PyG instead of PyTorch:
pyg-team/pytorch_geometric#2529 (comment)
https://pytorch-geometric.readthedocs.io/en/latest/modules/nn.html?highlight=DataParallel#module-torch_geometric.nn.data_parallel