Search code examples
pythongpugpgpuscientific-computing

High-level library for GPGPU


I want to use GPU for scientific computation. I know CUDA and OpenCL, but is there more high-level library (if possible, in Python) for GPGPU?


Solution

  • There are lots of ways doing this depending on what kinds of computations you want to do and how deeply you're willing to dive into the guts of the GPU.

    If you're using an NVIDIA GPU with CUDA, you can use the NVBLAS library supplied with the CUDA toolkit as a drop-in replacement that accelerates Level 3 BLAS functions.

    The Anaconda distribution of Python has a package called Accelerate that offers a bunch of features for GPU acceleration using NVIDIA GPU's.

    The pycuda package is another option.