python 矩阵运算 gpu_(cupy,minpy,mars,numba)使用GPU,并行计算和编译优化加速矩阵运算...

使用GPU，并行计算和编译优化加速numpy矩阵运算(相关材料整理)

v1(主要针对numpy运算的加速) 2020/12/16

总结：基于GPU加速numpy：cupy 和 minpy

​ 基于编译的优化加速numpy：numba

​ 基于并行计算加速numpy：Mars

​ 既可以并行又可以用GPU：Mars

numpy学习网址：

https://numpy.net/

​ https://www.numpy.org.cn/

​ http://cs231n.stanford.edu/syllabus.html

cupy

cupy支持使用GPU来加速Numpy。

cupy documents：https://docs.cupy.dev/en/stable/

如果已经安装好cuda，安装cupy只需要(安装之前一定要保证pip更新到最新的版本)

$ pip install cupy

也可以使用下面这个方法安装，

#根据自己安装的cuda版本是哪一个，然后直接下载安装适合的版本，实测这个方法比较快

#然后进行安装命令

# CUDA 8.0

pip install cupy-cuda80

# CUDA 9.0

pip install cupy-cuda90

# CUDA 9.1

pip install cupy-cuda91

# CUDA 9.2

pip install cupy-cuda92

# CUDA 10.0

pip install cupy-cuda1