pytorch相关函数在代码运行时速度超慢

pytorch相关函数在代码运行时速度超慢

最近在使用pytorch训练网络时遇到了一个很奇怪的问题。
我生成了一个(90000, 4, 4)的矩阵，然后调用torch.det()和torch.svd()，结果电脑直接卡住。
将矩阵大小减少到(10000, 4, 4)的时候也花费了近4秒才完成计算。
但是在命令行中使用torch.rand(90000, 4, 4)生成随机矩阵测试运行速度是正常的。
经过检查我自己的矩阵元素一切正常，没有奇异的情况。
尝试使用torch.save()保存到本地，然后在命令行中读取并使用pytorch计算，结果速度仍然很慢。
最后灵光一闪，发现测试数据是在cpu上的，我的代码是在gpu上的。
于是在代码中加入了.cpu()并运行，结果速度正常。
猜测是空余显存不够，pytorch并行效率下降导致的。
如果显存不太够用的情况下，有些计算步骤还是使用cpu计算吧