search

shenweichen / GraphEmbedding

Implementation and experiments of graph embedding algorithms.
MIT License
3.64k stars 990 forks source link

如何可以使用keras的gpu版本加速跑 #23

Closed ArlanCooper closed 3 years ago

ArlanCooper commented 4 years ago

我看现在是使用cpu跑的，速度非常的慢，如何使用GPU加速呢

shenweichen commented 4 years ago

安装gpu版的tensroflow就可以了

ArlanCooper commented 4 years ago

好的，谢谢