search

DeepGraphLearning / KnowledgeGraphEmbedding

MIT License
1.24k stars 264 forks source link

hi，a priblem about multi GPU training #46

Closed Zhangs996 closed 3 years ago

Zhangs996 commented 3 years ago

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

Edward-Sun commented 3 years ago

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