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

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

Gensimが提供するWord2vecとDoc2vecメソッドを使用して、BLAS、ATLASなどを使用して高速化する分散バージョンがあります(詳細ここ)。ただし、GPU モードはサポートされていますか? Gensim を使用すると GPU を動作させることは可能ですか?

ベストアンサー1

ご質問ありがとうございます。GPU の使用は Gensim のロードマップに含まれています。これに関してご意見があれば、ぜひお聞かせください。

@niitsumaによるkeras上で動作するword2vecのバージョンがあります。ワード2ベケラス最新のKerasバージョンで動作するコードは、このフォークとブランチにあります。https://github.com/SimonPavlik/word2vec-keras-in-gensim/tree/keras106

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

よろしくレブ

おすすめ記事