La bibliothèque Gensim prend-elle en charge l'accélération GPU?
En utilisant les méthodes Word2vec et Doc2vec fournies par Gensim, ils ont une version distribuée qui utilise BLAS, ATLAS, etc. pour accélérer (détails ici ). Cependant, prend-il en charge le mode GPU? Est-il possible de faire fonctionner le GPU si vous utilisez Gensim?
Merci pour votre question. L'utilisation du GPU figure sur la feuille de route de Gensim. Appréciera toute contribution que vous avez à ce sujet.
Il existe une version de Word2vec fonctionnant sur les kéros par @niitsuma appelée Word2veckeras . Le code qui s'exécute sur la dernière version de Keras se trouve dans cette branche et branche https://github.com/SimonPavlik/Word2vec-keras-in-gensim/tree/keras106
@SimonPavlik a exécuté un test de performances sur ce code. Il a constaté qu'un seul processeur graphique est plus lent que plusieurs processeurs pour Word2vec.
Cordialement Lev