Angel-ML / PyTorch-On-Angel

PyTorch On Angel, arming PyTorch with a powerful Parameter Server, which enable PyTorch to train very big models.
165 stars 51 forks source link

请问如何在GCN训练中设置使用GPU? #16

Closed AugF closed 3 years ago

ouyangwen-it commented 3 years ago

使用branch-gpu分支就可以了