Skip to main content

Gnumpy is a simple Python module that interfaces in a way almost identical to numpy, but does its computations on your computer's GPU, using Cudamat.

Project description

[This document is a copy of the original. The latest version is available on Tijmen Tieleman’s homepage.]

Gnumpy is free software, but if you use it in scientific work that gets published, you should cite this tech report in your publication.

Documentation: here.

Do you want to have both the compute power of GPU’s and the programming convenience of Python numpy? Gnumpy + Cudamat will bring you that.

Gnumpy is a simple Python module that interfaces in a way almost identical to numpy, but does its computations on your computer’s GPU. See this example, training an RBM using Gnumpy.

Gnumpy runs on top of, and therefore requires, the excellent cudamat library, written by Vlad Mnih.

Gnumpy can run in simulation mode: everything happens on the CPU, but the interface is the same. This can be helpful if you like to write your programs on your GPU-less laptop before running them on a GPU-equipped machine. It also allows you to easily test what performance gain you get from using a GPU. The simulation mode requires npmat, written by Ilya Sutskever. [npmat is included in this distribution.]

Gnumpy is licensed with a BSD-style license (i.e. it’s completely free to use for everyone, also as a component in commercial software), with one added note: if you use it for scientific work that gets published, you must include reference to the Gnumpy tech report in your publication. For details of the license, see the top of gnumpy.py.

Recent changes:

  • 2012-07-25: Bugfix. gnumpy.dot(x, x), when x is a 1-dimensional array, didn’t work but now works.

  • 2011-06-06: gnumpy.dot() now takes arrays of ndim>2.

  • 2011-04-19: Bugfix: several bugs involving zero size arrays were fixed.

  • 2011-04-15: Bugfix. “x=gnumpy.zeros(10); x[array([])] = garray([])” didn’t work as it should. Now it does.

  • 2011-03-24: Added gnumpy.outer().

  • 2011-03-15: The ability to check for infs and nans automatically has been added to Gnumpy.

  • 2010-07-19: Cudamat now enables fast indexing with arrays of indices. Download the newest Cudamat to have fast indexing with arrays in Gnumpy.

  • 2010-07-08: Renamed the project to Gnumpy. It used to be called Gpunnumpy.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

gnumpy-0.2.tar.gz (29.0 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page