3
Gensim에서 제공하는 Word2vec 및 Doc2vec 메서드를 사용하여 속도 향상을 위해 BLAS, ATLAS 등을 사용하는 분산 버전이 있습니다 (세부 사항은 here). 그러나 GPU 모드를 지원합니까? Gensim을 사용한다면 GPU를 작동시킬 수 있습니까?Gensim 라이브러리는 GPU 가속을 지원합니까?
A
답변
6
질문 해 주셔서 감사합니다. GPU 사용은 Gensim 로드맵에 있습니다. 당신은 그것에 대해 가지고있는 모든 의견을 주셔서 감사합니다.
word2veckeras이라는 @niitsuma에 의해 keras에서 실행되는 word2vec 버전이 있습니다. 최신 Keras 버전에서 실행되는 코드는이 분기 및 분기에 있습니다. https://github.com/SimonPavlik/word2vec-keras-in-gensim/tree/keras106
@SimonPavlik는이 코드에서 성능 테스트를 실행했습니다. 그는 하나의 GPU가 word2vec의 다중 CPU보다 느리다는 것을 발견했습니다.
감사 BLAS 그러나 GPU를 사용하지 않는
+0
안녕 레프, 링크에 대한 감사라는있다! Keras 버전이 있음을 아는 것은 좋은 일입니다. (여전히 Gensim을 사용하고 있지만). 어쩌면 다른 word2vec/doc2vec 버전도 검색 할 수 있습니다. 어쨌든, 앞으로 GPU를 기대 Gensim : D 고마워요 아이린 –
+3
안녕하세요 @ Lev이게 뭐야? 이것이 어떤 종류의 성능 개선을 가져올 것입니까? – user1712447
레프는 NVIDIA가 개발 한 GPU 버전이 CUBLAS – codetiger