векторы слов с использованием реализации gensim word2vec и графического процессора не показывают никакого ускорения

У меня компьютер с графическим процессором NVIDIA. Я установил OpenBLAS. Я пытаюсь обучить векторы слов, используя реализацию word2vec от gensim. Я установил количество рабочих = 4. Но когда я запускаю команду top, чтобы увидеть использование процессора. Показывает только 100%. Означает ли это, что используется только одно ядро? И моя программа не показывает разгона.

Мой фрагмент кода:

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)

person LCP    schedule 20.03.2016    source источник


Ответы (1)


Gensim в настоящее время не поддерживает использование графических процессоров: https://github.com/RaRe-Technologies/gensim/issues/449

person Kamil Sindi    schedule 02.12.2016