OpenCL integration for Python, plus shiny features

README.rst

PyOpenCL: Pythonic Access to OpenCL, with Arrays and Algorithms

https://badge.fury.io/py/pyopencl.png

PyOpenCL lets you access GPUs and other massively parallel compute devices from Python. It tries to offer computing goodness in the spirit of its sister project PyCUDA:

  • Object cleanup tied to lifetime of objects. This idiom, often called RAII in C++, makes it much easier to write correct, leak- and crash-free code.
  • Completeness. PyOpenCL puts the full power of OpenCL's API at your disposal, if you wish. Every obscure get_info() query and all CL calls are accessible.
  • Automatic Error Checking. All CL errors are automatically translated into Python exceptions.
  • Speed. PyOpenCL's base layer is written in C++, so all the niceties above are virtually free.
  • Helpful and complete Documentation as well as a Wiki.
  • Liberal license. PyOpenCL is open-source under the MIT license and free for commercial, academic, and private use.
  • Broad support. PyOpenCL was tested and works with Apple's, AMD's, and Nvidia's CL implementations.

Simple 4-step install instructions using Conda on Linux and macOS (that also install a working OpenCL implementation!) can be found in the documentation.

What you'll need if you do not want to use the convenient instructions above and instead build from source:

  • gcc/g++ new enough to be compatible with pybind11 (see their FAQ)
  • numpy, and
  • an OpenCL implementation. (See this howto for how to get one.)

Places on the web related to PyOpenCL: