Python GPU programming

python cuda gpu
Eelco Hoogendoorn picture Eelco Hoogendoorn · May 11, 2011 · Viewed 78.7k times · Source

I am currently working on a project in python, and I would like to make use of the GPU for some calculations.

At first glance it seems like there are many tools available; at second glance, I feel like im missing something.

Copperhead looks awesome but has not yet been released. It would appear that im limited to writing low-level CUDA or openCL kernels; no thrust, no cudpp. If id like to have something sorted, im going to have to do it myself.

That doesnt seem quite right to me. Am I indeed missing something? Or is this GPU-scripting not quite living up to the hype yet?

Edit: GPULIB seems like it might be what I need. Documentation is rudimentary, and the python bindings are mentioned only in passing, but im applying for a download link right now. Anyone has experience with that, or links to similar free-for-academic-use GPU libraries? ReEdit: ok, python bindings are infact nonexistant.

Edit2: So I guess my best bet is to write something in C/CUDA and call that from python?

Answer

Joseph Lisee picture Joseph Lisee · May 11, 2011

PyCUDA provides very good integration with CUDA and has several helper interfaces to make writing CUDA code easier than in the straight C api. Here is an example from the Wiki which does a 2D FFT without needing any C code at all.

Related questions

Cuda: library nvvm not found

I am trying to run the code below but an error is reported: NvvmSupportError: libNVVM cannot be found. Do conda install cudatoolkit: library nvvm not found My development environment is: Ubuntu 17.04, Spyder/Python3.5 and I have installed via conda (numba …

Read More
How to install CUDA in Google Colab GPU's

It seems that Google Colab GPU's doesn't come with CUDA Toolkit, how can I install CUDA in Google Colab GPU's. I am getting this error in installing mxnet in Google Colab. Installing collected packages: mxnet Successfully installed mxnet-1.2.0 ERROR: Incomplete …

Read More
trying to install pycuda, getting zip error?

windows 10, python 2.7 64 bit hello, following a guide to this step : pip install pipwin pipwin install pycuda gives me those options Package `pycuda` found in cache Choose version to download. [0] : 2014.1+cuda6514 [1] : 2015.1.3+cuda7518 no matter what i choose , i get the following …

Read More