centernet-pytorch icon indicating copy to clipboard operation
centernet-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

内存不足

Open tupig-7 opened this issue 4 years ago • 4 comments
trafficstars

使用16G内存训练，一会就内存不足了，是有什么地方需要设置吗？

tupig-7 avatar May 29 '21 01:05 tupig-7

你确认是内存不足不是显存不足？

bubbliiiing avatar May 30 '21 03:05 bubbliiiing

是的，内存不足。环境： 显卡：1080Ti 内存：16GB CPU：AMD3700X 使用num_worker=8,4,2运行一会都会内存不足，于是就没有用，这样不会超出内存，但是很慢，显卡占用不高，大概就是2GB

tupig-7 avatar May 30 '21 03:05 tupig-7

如果想提高一些速度的话可以调小一点点batch然后提高num_worker

bubbliiiing avatar Jun 05 '21 14:06 bubbliiiing

win系统下好像是pytorch的问题，num_worker设置为0试试看

Ferdinand-Byte avatar Oct 13 '21 05:10 Ferdinand-Byte