search

Henry1iu / TNT-Trajectory-Prediction

A Pytorch Implementation of TNT: Target-driveN Trajectory Prediction
487 stars 92 forks source link

运行时崩溃 #49

Open ItsPeter999 opened 1 year ago

ItsPeter999 commented 1 year ago

这个项目我在ubuntu（i7，32g内存，3060ti）上面跑，我只选了4w条数据，内存占用特别高，读完训练验证集32G内存就满了，然后就崩溃。但是在macbook（m1，8g内存）上就可以跑，而且不卡，只是训练得慢一点。

数据量是4000左右的时候，ubuntu还是能跑，挺快的，内存在用在12-16G之间。

这是为什么呀，有什么解决办法吗？

rjycty commented 1 year ago

这个模型需要的内存非常大，作者也写了需要128G+的内存，4w条数据32G肯定是不够的的，我5W条数据64G内存都会out of memory。 mac我没试过，不过作者给了个时间换内存的方法，你可以试下效果https://github.com/Henry1iu/TNT-Trajectory-Prediction/issues/11

ItsPeter999 commented 1 year ago

这个模型需要的内存非常大，作者也写了需要128G+的内存，4w条数据32G肯定是不够的的，我5W条数据64G内存都会out of memory。 mac我没试过，不过作者给了个时间换内存的方法，你可以试下效果#11

谢谢你的回答。的确是这样的，我去研究一下。

rjycty commented 1 year ago

GPU显存的消耗你有研究过吗，我现在用的2张16G显存的显卡，5W数据显示HIP out of memory