Seems unable to utilize multiple GPUs

[slnxyr]

Hi there.

I have tried running this code on one of my machine with two RTX3090 GPUs (GPU memory 24GB for each)

python -m torch.distributed.launch --nproc_per_node=2 script/run.py -c config/fb15k237.yaml --gpus [0,1]

However, I encountered the error saying that incompatible function arguments.

Could you help me solve this problem? Is there something wrong with my setups?

[KiddoZhu]

Hi! I guess this is because the graph data structure in TorchDrug is not compatible with the latest DDP implementation in PyTorch. What's your PyTorch version? Could you try PyTorch versions like 1.8.0 or 1.10.0?