What about the lowest GPU needed?

malllabiisc / CompGCN

ICLR 2020: Composition-Based Multi-Relational Graph Convolutional Networks

Apache License 2.0

597 stars 107 forks source link

What about the lowest GPU needed? #13

Closed leiloong closed 4 years ago

leiloong commented 4 years ago

I want to run the example given in Readme.md, but the GPU failed. So I want to the lowest GPU needed. It seems the GPU memory is enough, but it leads to black screen after training some epochs. Currently, I'm not sure if it is due to my GPU is not stable enough. Thanks.

svjan5 commented 4 years ago

Hi @leiloong, We trained all the models on Nvidia GeForce 1080Ti.