【python】GPU计算
2022/1/16 17:11:09
本文主要是介绍【python】GPU计算,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
A=cuda.to_device(a)
fgriddim, blockdim
b=A.copy_to_host()
cuda{
blockDim:[x],
blockIdx:[x],
threadDim:[x], threadIdx:[x], gridDim:[x],
}
from numba import cuda import numpy as np @cuda.jit def addGPU(A,B,C): i=cuda.blockIdx.x j=cuda.threadIdx.x C[i][j]=A[i][j]+B[i][j] @cuda.jit def mulGPU(A,B,C,k): i=cuda.blockIdx.x j=cuda.threadIdx.x C[i][j]=0 for k0 in range(k): C[i][j]=C[i][j]+A[i][k0]*B[k0][j] if __name__ == '__main__': print("start") N=3 A=np.array([ [1,0,0], [0,2,0], [0,0,1] ]) B=np.array([ [1,0,0], [0,1,1], [0,1,1] ]) d_A = cuda.to_device(A) d_B = cuda.to_device(B) d_C = cuda.to_device(np.zeros((N,N))) print("A:\n",A,"\nB:\n",B) addGPU[N, N](d_A,d_B,d_C) C = d_C.copy_to_host() print("add:\n",C) mulGPU[N, N](d_A,d_B,d_C,3) C = d_C.copy_to_host() print("mul:\n",C)
参考:https://blog.csdn.net/sazass/article/details/108649728
这篇关于【python】GPU计算的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-05-08有遇到过吗?同样的规则 Excel 中 比Python 结果大
- 2024-03-30开始python成长之路
- 2024-03-29python optparse
- 2024-03-29python map 函数
- 2024-03-20invalid format specifier python
- 2024-03-18pool.map python
- 2024-03-18threads in python
- 2024-03-14python Ai 应用开发基础训练,字符串,字典,文件
- 2024-03-13id3 algorithm python
- 2024-03-13sum array elements python