python使用skcuda包实现GPU上的矩阵乘法,矩阵求逆

导入需要的包:

import pycuda.gpuarray as gpuarray
import pycuda.autoinit
import skcuda.linalg as sklin

创建两个矩阵:

a = numpy.random.randn(4, 4).astype(numpy.float32) #为了后续GPU上的计算顺利进行,矩阵数值设定为float32
b = numpy.random.randn(4, 4).astype(numpy.float32)

使用gpuarray将矩阵转换为GPU下可用的类型:

a_gpu = gpuarray.to_gpu(a)
b_gpu = gpuarray.to_gpu(b)

使用skcuda的线性代数库实现矩阵乘法:

sklin.init()   # 使用skcuda线性代数库时要先调用内置函数进行初始化
multi_gpu = sklin.dot(a_gpu, b_gpu)  # 矩阵乘法
a_inv = sklin.inv(a_gpu)  # 矩阵求逆
multi = multi_gpu.get()
inv = a_inv.get()
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值