minpy使用GPU加速Numpy科學計算

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

然後就可以正常使用了。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章