Home

Awesome

CUDA-based NumPy

CUDArray is a CUDA-accelerated subset of the NumPy library. The goal of CUDArray is to combine the easy of development from the NumPy with the computational power of Nvidia GPUs in a lightweight and extensible framework.

CUDArray currently imposes many limitations in order to span a manageable subset of the NumPy library. Nonetheless, it supports a neural network pipeline as demonstrated in the project deeppy.

Features

Installation

With CUDA back-end

First, you should consider specifying the following environment variables.

Then build and install libcudarray with

make
make install

Finally, install the cudarray Python package:

python setup.py install
Without CUDA back-end

Install the cudarray Python package:

python setup.py --without-cuda install

Documentation

Please consult the technical report for now. Proper documentation is on the TODO list.

Contact

Feel free to report an issue for feature requests and bug reports.

For a more informal chat, visit #cudarray on the freenode IRC network.

Citation

If you use CUDArray for research, please cite the technical report:

@techreport{larsen2014cudarray,
  author = "Larsen, Anders Boesen Lindbo",
  title = "{CUDArray}: {CUDA}-based {NumPy}",
  institution = "Department of Applied Mathematics and Computer Science, Technical University of Denmark",
  year = "2014",
  number = "DTU Compute 2014-21",
}

TODO

Influences

Thanks to the following projects for inspiration.