文章/答案/技术大牛

发布

社区首页 >问答首页 >Gensim库是否支持GPU加速？

问Gensim库是否支持GPU加速？
EN

Stack Overflow用户

提问于 2016-09-18 22:20:49

回答 1查看 16.4K关注 0票数 20

使用Gensim提供的Word2vec和Doc2vec方法，他们有一个使用BLAS、ATLAS等来加速的分布式版本(details here)。但是，它是否支持GPU模式？如果使用Gensim，可以让GPU正常工作吗？

发布于 2016-09-29 12:45:34

谢谢你的问题。使用GPU在Gensim路线图上。我将非常感谢您对它的任何意见。

@niitsuma在keras上运行了一个名为word2veckeras的word2vec版本。在最新版本的Keras上运行的代码位于此分支和分支https://github.com/SimonPavlik/word2vec-keras-in-gensim/tree/keras106中

@SimonPavlik已经对此代码运行了性能测试。他发现，对于word2vec来说，单个图形处理器比多个gpu慢。

关注Lev

票数 20

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/39558642

复制

相似问题

问Gensim库是否支持GPU加速？EN