CUDA内存与矩阵相乘

作者: 大黑骡子王
发布时间: 2024-12-16 06:46:56 (6天前)
转自：

2 条回复

0#
回复此人
电动少女 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 乘以矩阵，您的输出大小将为3,000 x 3,000,000矩阵！尽管如此 <code> A </code> 和 <code> B </code> 相对较小的输出 <code> R </code> 是巨大的：9 的<strong> G </强> 元素。而且，我怀疑 <code> dtype </code> 你的矩阵是 <code> float64 </code> 并不是 <code> float32 </code> （因为你使用numpy来初始化它们）。因此，每个9G元素 <code> R_gpu </code> 需要8个字节;带你到至少72的大小的<strong> GB </强> GPU内存仅适用于 <code> R_gpu </code> 。我怀疑中间结果和其他一些东西占用了你的GPU内存。 </p> </DIV>

编辑

登录后才能参与评论