Ire*_* Li 12 optimization gpu gensim deeplearning4j
使用Gensim提供的Word2vec和Doc2vec方法,他们有一个使用BLAS,ATLAS等加速的分布式版本(详情请参见此处).但是,它支持GPU模式吗？如果使用Gensim,是否可以让GPU工作？
小智 15
谢谢你的问题.使用GPU是在Gensim路线图上.将欣赏您的任何输入.
有一个版本的word2vec由@niitsuma在keras上运行,名为word2veckeras.在最新的Keras版本上运行的代码位于此分支和分支中https://github.com/SimonPavlik/word2vec-keras-in-gensim/tree/keras106
@SimonPavlik对此代码进行了性能测试.他发现word2vec的单个gpu比多个CPU慢.
关心列夫