web-dev-qa-db-ja.com

GensimライブラリはGPUアクセラレーションをサポートしていますか?

Gensimが提供するWord2vecおよびDoc2vecメソッドを使用すると、BLAS、ATLASなどを使用してスピードアップする分散バージョンがあります(詳細 ここ )。ただし、GPUモードをサポートしていますか? Gensimを使用している場合、GPUを機能させることはできますか?

15
Irene Li

ご質問ありがとうございます。 GPUの使用はGensimロードマップにあります。あなたがそれについて持っているどんな入力にも感謝します。

@niitsumaによってkerasで実行されているWord2vecのバージョンは Word2veckeras と呼ばれています。最新のKerasバージョンで実行されるコードはこのフォークとブランチにあります https://github.com/SimonPavlik/Word2vec-keras-in-gensim/tree/keras106

@SimonPavlikはこのコードでパフォーマンステストを実行しました。彼は、Word2vecの単一のCPUが複数のCPUよりも遅いことを発見しました。

よろしくレヴ

17
Lev Konst