Wort Vektoren mit Gensim's Word2vec-Implementierung und GPU zeigt keine Beschleunigung

Question

Ich habe PC mit NVIDIA GPU. Ich habe OpenBLAS installiert. Ich versuche, Wortvektoren mit Gensim's Word2vec-Implementierung zu trainieren. Ich habe die Anzahl der Arbeiter = 4 festgelegt. Aber wenn ich Top-Befehl ausführen, um die CPU-Auslastung zu sehen. Es zeigt nur 100%. Bedeutet es, dass nur ein Kern verwendet wird? Und mein Programm zeigt keine Beschleunigung.

Mein Code-Schnipsel ist:

import gensim 
import time 
import numpy 
class MySentences(object): 
    def __init__(self, dirname): 
     self.dirname = dirname 
    #called when Word2Vec is called 
    def __iter__(self): 
     for fname in os.listdir(self.dirname): 
      for line in open(os.path.join(self.dirname, fname)): 
       yield line.split() 


sentences=MySentences("/home/lalchand/NewdatasetforAssgn2/tfidf/spam") 


start = time.time() 

model = gensim.models.Word2Vec(sentences, min_count=1,iter=5,workers=4) 
print(model.syn0.shape) 

Answer 1

GENSIM derzeit nicht mit GPUs unterstützen: https://github.com/RaRe-Technologies/gensim/issues/449