[Numpy-discussion] Direct GPU support on NumPy

Lev E Givon lev at columbia.edu
Tue Jan 2 20:56:21 EST 2018


On Jan 2, 2018 8:35 PM, "Matthew Harrigan" <harrigan.matthew at gmail.com>
wrote:

Is it possible to have NumPy use a BLAS/LAPACK library that is GPU
accelerated for certain problems?  Any recommendations or readme's on how
that might be set up?  The other packages are nice but I would really love
to just use scipy/sklearn and have decompositions, factorizations, etc for
big matrices go a little faster without recoding the algorithms.  Thanks


Depending on what operation you want to accelerate, scikit-cuda may provide
a scipy-like interface to GPU-based implementations that you can use. It
isn't a drop-in replacement for numpy/scipy, however.

http://github.com/lebedov/scikit-cuda

L
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/numpy-discussion/attachments/20180102/29616d57/attachment.html>


More information about the NumPy-Discussion mailing list