DeepGraphLearning/KnowledgeGraphEmbedding

hi，a priblem about multi GPU training

Zhangs996 opened this issue · 1 comments

Zhangs996 commented

how can i change the code to do multi gpu training
怎么样才能用多个gpu进行训练呢，我用的我自己的数据集，报的是显存溢出，我的显存40G，数据量太大啦，请问可以怎么修改呢？谢谢

Edward-Sun commented

Please check GraphVite (https://github.com/DeepGraphLearning/graphvite) for large-scale multi-GPU training.