PyCUDA - это модуль Python, который обеспечивает комплексный питонический интерфейс для вычислительной среды NVIDIA CUDA GPU.
PyCUDA предоставляет модуль python для доступа к API драйвера NVIDIA CUDA из кода Python.
Модуль включает взаимодействие с numpy и комплексные средства метапрограммирования для динамической генерации и JIT-компиляции кода CUDA с использованием Python.