侧边栏壁纸
博主头像
乌拉队长博主等级

你只管努力,其余的交给命运

  • 累计撰写 122 篇文章
  • 累计创建 34 个标签
  • 累计收到 31 条评论

目 录CONTENT

文章目录

minpy使用GPU加速Numpy科学计算

乌拉队长
2020-02-25 / 0 评论 / 0 点赞 / 3,130 阅读 / 0 字

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

然后就可以正常使用了。

0

评论区