I build a gnn model. It can run in one gpu. But when I deploy it on multiple gpus using ‘model = nn.DataParallel(model, device_ids=[0,1,2,3])’, I will get an error, AttributeError: ‘NoneType’ object has no attribute ‘varcount’. My gnn model :https://paste.ubuntu.com/p/8Pr4f4Wz3v/
Hi there,
Currently multi GPU support is not yet tested in 0.1 versions. It is right now being worked on in 0.2.
You are welcome to push a fix too!
Thank you.