Optimization Gensim库是否支持GPU加速？

使用Gensim提供的Word2vec和Doc2vec方法，他们有一个分布式版本，使用BLAS、ATLAS等来加速（详细信息）。但是，它是否支持GPU模式？如果使用Gensim，GPU是否可以正常工作？

谢谢您的提问。使用GPU是Gensim的路线图。如果您对此有任何意见，我们将不胜感激

有一个版本的word2vec运行在keras上，由@niitsuma调用。 在最新Keras版本上运行的代码在此分支中

@SimonPavlik已经对此代码运行了性能测试。他发现word2vec的单个gpu比多个CPU慢

问候

---

Lev

BLAS不使用GPU，但是，NVIDIA开发了一个名为cuBLASHi Lev的GPU版本，感谢链接！很高兴知道有一个Keras版本（尽管我想它仍然在使用Gensim）。也许我也可以搜索其他word2vec/doc2vec版本。不管怎样，期待GPU Gensim:D感谢IreneHi@Lev这是什么现状？这将带来什么样的性能改进？问题是：Word2VEC使用GPUE需要考虑可重复性结果的相关性。使用FastText可以只使用一个线程。我相信凯拉斯也有同样的问题。但是使用Pytork可以复制与设置火炬完全相同的训练过程。此命令将影响所有设备（CPU或GPU），但某些命令可能插入随机行为。