python(numpy)能处理15000x50*50x850000大小的矩阵乘法吗?

2024-10-03 19:33:20 发布

您现在位置:Python中文网/ 问答频道 /正文

设A为尺寸为150000x50的矩阵,B为尺寸为85000x50的矩阵。我想执行乘法A*B.T,我认为numpy在处理这种规模时不会有问题(即使需要几分钟)。但是,约30分钟后操作失败。我在一台强大的机器上运行——GCP16vCPU,64GB内存

有没有人有过执行这种规模的matmul的经验?是否有一个变通办法使其发挥作用?一些提供并行化的库


Tags: 内存numpy机器尺寸矩阵经验规模乘法