Does Gensim library support GPU acceleration?

Question

Using Word2vec and Doc2vec methods provided by Gensim, they have a distributed version which uses BLAS, ATLAS, etc to speedup (details here). However, is it supporting GPU mode? Is it possible to get GPU working if using Gensim?

Answer 1

Thank you for your question. Using GPU is on the Gensim roadmap. Will appreciate any input that you have about it.

There is a version of word2vec running on keras by @niitsuma called word2veckeras. The code that runs on latest Keras version is in this fork and branch https://github.com/SimonPavlik/word2vec-keras-in-gensim/tree/keras106

@SimonPavlik has run performance test on this code. He found that a single gpu is slower than multiple CPUs for word2vec.

Regards Lev