pymc3: параллельные вычисления с njobs> 1 против GPU

Я пытаюсь ускорить выборку pymc3 с параллелизацией, и я вижу только скромную выгоду.pymc3: параллельные вычисления с njobs> 1 против GPU

Мне удалось уменьшить общее время работы от 25 минут (njobs = 1) до 13 минут (njobs = 6) на i7 MacBook Pro. В связи с тем, что для начала выборки требуется около 4 минут, прежде чем pymc начнет выборку, увеличение относительно невелико.

Вопрос в том, кто-нибудь успешно использует GPU с pymc3 и какую выгоду я могу получить для моделей, которые занимают 6-8 минут? (У моего MacBook есть nvidia GT 750M 2Gb)

источник

2014-12-09 volodymyr

Я запускаю Linux на Intel i7-4930.

Я побежал модель PyMC3, что потребовалось 90 минут на CPU (с использованием всех ядер), но занял только 18 минут на моем GeForce GTX 970.

Так ускорение почти в 5 раз.

источник

2015-07-23 00:25:25 inversion

Не могли бы вы разместить образец кода, показывающий это ускорение? Я играл с примерами в дистрибутиве, и я всегда получаю то же самое время, будь то использование процессора или графического процессора ... –

pymc3: параллельные вычисления с njobs> 1 против GPU

ответ

Смежные вопросы