gpt4 book ai didi

optimization - Gensim库是否支持GPU加速?

转载 作者:行者123 更新时间:2023-12-03 15:29:28 25 4
gpt4 key购买 nike

使用Gensim提供的Word2vec和Doc2vec方法,它们具有使用BLAS,ATLAS等进行加速的分布式版本(详细信息here)。但是,它支持GPU模式吗?如果使用Gensim,是否可以使GPU工作?

最佳答案

谢谢你的问题。 Gensim路线图上正在使用GPU。将不胜感激您对此的任何投入。

@niitsuma在keras上运行的word2vec版本称为word2veckeras
在最新的Keras版本上运行的代码在此fork和分支https://github.com/SimonPavlik/word2vec-keras-in-gensim/tree/keras106

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

问候
列夫

关于optimization - Gensim库是否支持GPU加速?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39558642/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com