if it is support torch.nn.parallel.DistributedDataParallel oneday #6156
STsingerST
started this conversation in
General
Replies: 1 comment
-
See pytorch_geometric/blob/master/examples/multi_gpu/distributed_sampling.py |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Now I am use torch_geometric.nn.DataParallel which is from torch.nn.DataParallel, but it is very slow even if use many GPUs.
If it is possible to support torch.nn.parallel.DistributedDataParallel oneday in PyG?
Beta Was this translation helpful? Give feedback.
All reactions