内存泄漏

post_img
发布于 2022-05-21
6439 热度 无~ Pytorch

记一次Pytorch内存泄漏排坑

摘要

最近用Pytorch跑模型，在自己电脑上跑的挺好的，但是一送到服务器上去跑就挂了。一开始以为是batchsize开太大了爆炸了，后 …