[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>

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.


-------------- 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