minpy使用GPU加速Numpy科学计算

minpy使用GPU加速Numpy科学计算

Scroll Down

minpy是一个基于MXNet的支持GPU的加速Numpy计算的库,用法和Numpy几乎一样,使用:

import scipy.io as sio
import matplotlib as plt
import minpy.numpy as np

使用时只需要在Numpy前面加上minpy,就可以像Numpy一样使用它进行矩阵运算。

甩一条 MXNet官网链接

minpy安装起来也很简单:

先安装MXNet依赖

# 安装cuda10.1版本的MXNet
pip install mxnet-cu101


# 如果你的cuda版本为10.0,则执行下面的命令,其他版本同理
pip install mxnet-cu100

然后安装minpy:

pip install minpy

然后就可以正常使用了。