Gensim庫是否支持GPU加速？

使用Gensim提供的Word2vec和Doc2vec方法，他們有一個使用BLAS，ATLAS等加速的分佈式版本（詳細信息here）。但是，它支持GPU模式嗎？如果使用Gensim，可以使GPU工作嗎？Gensim庫是否支持GPU加速？

2016-09-18 Irene Li

BLAS不使用，無論GPU，有一個由NVIDIA開發的GPU版本，稱爲CUBLAS – codetiger

謝謝你的問題。使用GPU在Gensim路線圖上。希望您對此有任何意見。

@niitsuma在keras上運行的word2vec的版本號爲word2veckeras。在最新的Keras版本上運行的代碼位於此分支和分支中https://github.com/SimonPavlik/word2vec-keras-in-gensim/tree/keras106

@SimonPavlik已對此代碼執行性能測試。他發現單個gpu比word2vec的多個CPU要慢。

問候利

2016-09-29 04:45:33

喜利，感謝您的聯繫！很高興知道有一個Keras版本（儘管它仍然使用Gensim我想）。也許我可以搜索其他word2vec/doc2vec版本。無論如何，期待着GPU Gensim：D 感謝艾琳 –

嗨@Lev這是什麼現狀？這應該帶來什麼樣的性能改進？ – user1712447

回答