使用 Gensim 提供的 Word2vec 和 Doc2vec 方法，他们有一个分布式版本，使用 BLAS、ATLAS 等来加速（详见此处）。但是，它是否支持 GPU 模式？如果使用 Gensim，是否可以让 GPU 工作？
问问题
17175 次
1 回答
20
谢谢你的问题。使用 GPU 在 Gensim 路线图上。将感谢您对此的任何意见。
@niitsuma 在 keras 上运行了一个 word2vec 版本，名为word2veckeras。在最新 Keras 版本上运行的代码在这个分支和分支中https://github.com/SimonPavlik/word2vec-keras-in-gensim/tree/keras106
@SimonPavlik 对此代码进行了性能测试。他发现对于 word2vec，单个 gpu 比多个 CPU 慢。
问候列夫
于 2016-09-29T04:45:33.997 回答