使用Gensim提供的Word2vec和Doc2vec方法，他们有一个使用BLAS，ATLAS等加速的分布式版本（详情here）。但是，它支持GPU模式吗？如果使用Gensim，是否可以让GPU工作？
答案 0 :(得分：11)
感谢您的提问。使用GPU是在Gensim路线图上。非常感谢您对此有任何意见。
@ niitsuma在keras上运行了一个名为word2veckeras的word2vec版本。 在最新的Keras版本上运行的代码位于此分支https://github.com/SimonPavlik/word2vec-keras-in-gensim/tree/keras106
中@SimonPavlik已对此代码进行了性能测试。他发现word2vec的单个gpu比多个CPU慢。
此致 列夫