Научные вычисления на Python с современными графическими процессорами с двойной точностью

14

Кто-нибудь здесь использовал научные вычисления двойной точности с графическими процессорами нового поколения (например, K20) через Python?

Я знаю, что эта технология быстро развивается, но каков наилучший способ сделать это в настоящее время? GPU не входит в число популярных научных библиотек Python numpy и scipy, и я хотел использовать theano, но, похоже, он использует только точность float32 для GPU. Мне известно, что Google может предоставить результаты поиска для Python GPU, но я надеюсь на большее понимание, чем список проектов, которые могут или не могут встретиться с их создателем.

python gpu precision клипер
источник
4
Если бы мне это было нужно, я бы использовал PyOpenCL . Кодирование общего назначения на GPU все еще находится на довольно низком уровне (попробуйте интерфейс OpenCL C, это сложно). Тем не менее, PyOpenCL, по-видимому, абстрагируется как можно больше, и похоже, что за этим стоит значительный импульс.
boyfarrell
1
«Новое поколение» должно быть довольно похоже на более раннюю серию Tesla с двойной точностью, вероятно, единственной оговоркой является версия CUDA / драйвера. Поэтому методы двойной точности, которые работают с Tesla (например, M2070) и текущей версией CUDA / Driver, также должны работать с K20.
internetscooter
1
Может быть, здесь ( stackoverflow.com/questions/5957554/python-gpu-programming ) может помочь вам.
eusoubrasileiro
2
У Theano новый графический процессор, поддерживающий float64. Он еще не завершен, но мы объявим его в бета-версии на этой неделе.
Нуиз
Спасибо @nouiz - я бы предложил добавить ваш комментарий в качестве ответа при создании релиза.
Арон Ахмадиа

Ответы:

3

Я не знаю, почему я поставил этот ответ в комментарии ...

Если бы мне это было нужно, я бы использовал PyOpenCL . Кодирование общего назначения на GPU все еще находится на довольно низком уровне (попробуйте интерфейс OpenCL C, это сложно). Тем не менее, PyOpenCL, по-видимому, абстрагируется как можно больше, и похоже, что за этим стоит значительный импульс.

boyfarrell
источник
Эх, бывает. Я делаю то же самое иногда.
Джефф Оксберри