Можно ли использовать видеокарту для вычислений?

Можно ли использовать видеокарту вместо процессора для вычислений в python? Если да, то как?


Ответы (1 шт):

Автор решения: Nyūron

Вроде как поддержки CUDA у python нет, но есть обертка PyCUDA, если у Вас видеокарта от NVidia: https://documen.tician.de/pycuda/, либо NUMBA, как написали выше. Вот статтья на сравнение этих двух: https://habr.com/ru/post/317328/ И пособие в котором описано больше по GPU+Python: https://cmp.phys.msu.ru/sites/default/files/GPUPython.pdf

→ Ссылка