水木社区手机版
首页
|版面-数学工具(MathTools)|
新版wap站已上线
展开
|
楼主
|
同主题展开
|
溯源
|
返回
上一篇
|
下一篇
|
同主题上篇
|
同主题下篇
主题:Re: numpy超大矩阵相乘的问题
Blue
|
2022-10-15 15:39:35
|
昨天实验了GPU,用cupy来计算,一开始可高兴了,10000*10000矩阵的平方计算结果太快了,不到1秒,于是升级到20000*20000,挂了,out of memory,显存只有6G,两个这样的矩阵直接占用了5G的内存。唉,看来只有用numpy结合cupy,将矩阵掰碎到GPU中计算。
要是再大一个数量级的计算,可能只有去找spark了
--
FROM 114.253.3.*
上一篇
|
下一篇
|
同主题上篇
|
同主题下篇
选择讨论区
首页
|
分区
|
热推
BYR-Team
©
2010.
KBS Dev-Team
©
2011
登录完整版