2016-09-18 37 views
3

Gensimが提供するWord2vecおよびDoc2vecメソッドを使用すると、高速化のためにBLAS、ATLASなどを使用する分散バージョンがあります(詳細はhere)。しかし、それはGPUモードをサポートしていますか? Gensimを使用している場合、GPUを動作させることは可能ですか?GensimライブラリはGPUアクセラレーションをサポートしていますか?

+0

BLASは、しかし、GPUを使用しない、NVIDIAが開発したGPUのバージョンがあるが、リンクについて おかげで、 – codetiger

答えて

6

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

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

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

よろしく レフ

+0

こんにちはレフCUBLASと呼ばれます! Kerasのバージョンがあることを知っておいてよかったです(まだGensimを使用していますが)。多分私は他のword2vec/doc2vecのバージョンを検索することもできます。とにかく、GPUを楽しみにしてGensim:D ありがとうIrene –

+3

こんにちは@Levこれの現在の状態は何ですか?これはどのようなパフォーマンスの改善をもたらしますか? – user1712447