CUDA&numba | 使用python分别在cpu和gpu全局内存和gpu共享内存进行矩阵乘法运算

矩阵乘法

在这里插入图片描述

导入相关的包

from numba import cuda,float32
import numba
import numpy as np
import math
import time

在cpu上进行运算

猜你喜欢

转载自blog.csdn.net/weixin_43236007/article/details/109829521