Seems unable to utilize multiple GPUs

Open slnxyr opened this issue 3 years ago • 1 comments

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?

Oct 13 '22 05:10 slnxyr

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?

Oct 15 '22 22:10 KiddoZhu